set pytorch to 2.1 cuda 12.1

cyntachs · Oct 30, 2023 · e208adf · e208adf
1 parent 3b7175f
commit e208adf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,6 +1,6 @@
-FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-runtime
+FROM pytorch/pytorch:2.1.0-cuda12.1-cudnn8-runtime
 
-LABEL version="1.2-cuda11.8" maintainer="Cyntachs"
+LABEL version="1.2-cuda12.1" maintainer="Cyntachs"
 
 ARG DEBIAN_FRONTEND=noninteractive
 ENV TZ="America/New_York"