books/LLMs-from-scratch
1
0
Fork 0
mirror of https://github.com/rasbt/LLMs-from-scratch.git synced 2026-04-10 12:33:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #65 from d-kleine/main

Browse Source 
Updated Dockerfile
This commit is contained in:
Sebastian Raschka
2024-03-11 06:39:33 -05:00
committed by GitHub
parent e524ddb6f4 3787227c41
commit 8c1871f16e
1 changed files with 5 additions and 6 deletions
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