Update install_pytorch.sh

GregoryComer · web-flow · commit d51ecea736ad · 2025-10-13T14:49:49.000-07:00
diff --git a/.ci/docker/common/install_pytorch.sh b/.ci/docker/common/install_pytorch.sh
@@ -17,7 +17,7 @@ install_domains() {
 }
 
 install_pytorch_and_domains() {
-  pip_install torch==2.9.0 torchvision torchaudio torchao=0.14.0 --index-url https://download.pytorch.org/whl/test/cpu
+  pip_install torch==2.9.0 torchvision torchaudio torchao==0.14.0 --index-url https://download.pytorch.org/whl/test/cpu
 }
 
 install_pytorch_and_domains

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ install_domains() {`
`17`	`17`	`}`
`18`	`18`
`19`	`19`	`install_pytorch_and_domains() {`
`20`		`- pip_install torch==2.9.0 torchvision torchaudio torchao=0.14.0 --index-url https://download.pytorch.org/whl/test/cpu`
	`20`	`+ pip_install torch==2.9.0 torchvision torchaudio torchao==0.14.0 --index-url https://download.pytorch.org/whl/test/cpu`
`21`	`21`	`}`
`22`	`22`
`23`	`23`	`install_pytorch_and_domains`