[torchx][CI] Unpin torch, torchvision, torchxtext to fix breaking unittests due to incompatible version of torchvision/text and torch (#1147)

kiukchung · Kiuk Chung · web-flow · commit 426b91a5728a · 2025-10-16T16:01:02.000-07:00
Co-authored-by: Kiuk Chung &lt;kiuk@google.com&gt;
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -19,11 +19,11 @@ pytorch-lightning==2.5.0
 tensorboard==2.14.0
 sagemaker==2.230.0
 torch-model-archiver>=0.4.2
-torch>=2.7.0
+torch
 torchmetrics==1.6.3
 torchserve>=0.10.0
-torchtext==0.18.0
-torchvision==0.23.0
+torchtext
+torchvision
 typing-extensions
 ts==0.5.1
 wheel
diff --git a/torchx/runtime/container/Dockerfile b/torchx/runtime/container/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.9.0-cuda12.8-cudnn9-runtime
 
 WORKDIR /app
 

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime`
	`1`	`+FROM pytorch/pytorch:2.9.0-cuda12.8-cudnn9-runtime`
`2`	`2`
`3`	`3`	`WORKDIR /app`
`4`	`4`