|
|
@@ -4,12 +4,14 @@ FROM pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime
|
|
|
# Install Ubuntu packages
|
|
|
RUN apt-get update && \
|
|
|
apt-get upgrade -y && \
|
|
|
- apt-get install -y rsync && \
|
|
|
- apt-get install -y git && \
|
|
|
- apt-get install -y curl && \
|
|
|
+ apt-get install -y rsync git curl ca-certificates && \
|
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
|
|
+# Install uv
|
|
|
+ADD https://astral.sh/uv/install.sh /uv-installer.sh
|
|
|
+RUN sh /uv-installer.sh && rm /uv-installer.sh
|
|
|
+ENV PATH="/root/.local/bin/:$PATH"
|
|
|
+
|
|
|
# Install Python packages
|
|
|
COPY requirements.txt requirements.txt
|
|
|
-RUN pip install --upgrade pip
|
|
|
-RUN pip install --no-cache-dir -r requirements.txt
|
|
|
+RUN uv pip install --system --no-cache -r requirements.txt
|