Merge pull request #39 from rayedbw/patch-3

Update Dockerfile
This commit is contained in:
Sebastian Raschka 2024-02-29 12:30:50 -06:00 committed by GitHub
commit d419c02792

View File

@ -1,4 +1,4 @@
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-devel
FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
RUN <<EOF
apt-get update