mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2026-04-10 12:33:42 +00:00
13 lines
252 B
Docker
13 lines
252 B
Docker
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-devel
|
|
|
|
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
|