added uv to Dockerfile (#537)

* added uv to Dockerfile

* Update Dockerfile

---------

Co-authored-by: Sebastian Raschka <mail@sebastianraschka.com>
This commit is contained in:
Daniel Kleine 2025-02-18 21:53:19 +01:00 committed by GitHub
parent 0a5214b804
commit 58aabe7dd8

View File

@ -4,12 +4,14 @@ FROM pytorch/pytorch:2.5.0-cuda12.4-cudnn9-runtime
# Install Ubuntu packages # Install Ubuntu packages
RUN apt-get update && \ RUN apt-get update && \
apt-get upgrade -y && \ apt-get upgrade -y && \
apt-get install -y rsync && \ apt-get install -y rsync git curl ca-certificates && \
apt-get install -y git && \
apt-get install -y curl && \
rm -rf /var/lib/apt/lists/* 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 # Install Python packages
COPY requirements.txt requirements.txt COPY requirements.txt requirements.txt
RUN pip install --upgrade pip RUN uv pip install --system --no-cache -r requirements.txt
RUN pip install --no-cache-dir -r requirements.txt