updated pytorch to 2.2

updated to pytorch 2.2.2 with cuda 12.1
cyntachs · Apr 22, 2024 · e1e9f63 · e1e9f63
1 parent 8409638
commit e1e9f63
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-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
 
-LABEL version="1.2-cuda12.1" maintainer="Cyntachs"
+LABEL version="py2.2-cuda12.1" maintainer="Cyntachs"
 
 ARG DEBIAN_FRONTEND=noninteractive
 ENV TZ="America/New_York"