mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2026-04-10 12:33:42 +00:00
12 lines
248 B
Docker
12 lines
248 B
Docker
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
|
|
|
|
RUN <<EOF
|
|
apt-get update
|
|
apt-get install -y --no-install-recommends git
|
|
rm -rf /var/lib/apt/lists/*
|
|
EOF
|
|
|
|
COPY requirements.txt requirements.txt
|
|
|
|
RUN pip install --no-cache-dir -r requirements.txt
|