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

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