Update Dockerfile

Use significantly smaller docker image
This commit is contained in:
Rayed Bin Wahed
2024-03-01 02:10:01 +08:00
committed by GitHub
parent a94d53a752
commit 32087331ae

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