mirror of
https://github.com/rasbt/LLMs-from-scratch.git
synced 2026-04-10 12:33:42 +00:00
3787227c41
* changed CUDA files to pytorch 2.0.1 (for reproducibility) * fixed RUN command (for updating Ubuntu and installing Git)
11 lines
260 B
Docker
11 lines
260 B
Docker
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
|
|
|
RUN apt-get update && \
|
|
apt-get upgrade -y && \
|
|
apt-get install -y git && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
COPY requirements.txt requirements.txt
|
|
|
|
RUN pip install --no-cache-dir -r requirements.txt
|