mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2025-08-30 19:41:51 +00:00
58aabe7dd8
* added uv to Dockerfile * Update Dockerfile --------- Co-authored-by: Sebastian Raschka <mail@sebastianraschka.com>
18 lines
532 B
Docker
18 lines
532 B
Docker
# Install PyTorch 2.5 with CUDA 12.4
|
|
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 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 uv pip install --system --no-cache -r requirements.txt
|