upgrade pytorch to 2.2 (#827)

pytorch · Feb 17, 2024 · 6df4416 · 6df4416
1 parent 2845aec
commit 6df4416
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchx/runtime/container/Dockerfile b/torchx/runtime/container/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
 
 WORKDIR /app