yujunjun/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2025-12-12 15:31:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

Updated Dockerfile with following changes:

Browse Source 
* changed CUDA files to pytorch 2.0.1 (for reproducibility)
* fixed RUN command (for updating Ubuntu and installing Git)
This commit is contained in:
Daniel Kleine 2024-03-11 08:06:48 +00:00
parent 321f3d33f9
commit 3787227c41
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
.devcontainer/Dockerfile
View File

@ -1,10 +1,9 @@
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
RUN <<EOF
apt-get update
apt-get install -y --no-install-recommends git
rm -rf /var/lib/apt/lists/*
EOF
RUN apt-get update && \
apt-get upgrade -y && \
apt-get install -y git && \
rm -rf /var/lib/apt/lists/*
COPY requirements.txt requirements.txt