Merge pull request #21 from js-ish/feat-update-pytorch

fix: change cuda to 12.1.0
js-ish · May 10, 2024 · 53ae90c · 53ae90c
2 parents 7d55b64 + 4afe54c
commit 53ae90c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch/base.Dockerfile b/pytorch/base.Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.2.2-devel-ubuntu20.04
+FROM nvidia/cuda:12.1.0-devel-ubuntu20.04
 
 RUN set -eux; \
     apt-get update; \