docker: Upgrade pytorch to 2.0 (pytorch#743)

seemethere · MichaelClifford · commit 35556be0f77f · 2023-09-08T14:31:48.000-04:00
Signed-off-by: Eli Uriegas &lt;eliuriegas@meta.com&gt;
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
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM pytorch/pytorch:1.13.0-cuda11.6-cudnn8-runtime`
	`1`	`+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime`
`2`	`2`
`3`	`3`	`WORKDIR /app`
`4`	`4`