docker: Upgrade pytorch to 2.0

Signed-off-by: Eli Uriegas <[email protected]>
pytorch · Jul 26, 2023 · 1c960e1 · 1c960e1
1 parent 83268d6
commit 1c960e1
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:1.13.0-cuda11.6-cudnn8-runtime
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 WORKDIR /app