fix(api): revert torch to 2.0.1 for ROCm due to MIopen errors (#444)

ssube · Dec 22, 2023 · 0aab130 · 0aab130
1 parent ee35737
commit 0aab130
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/api/requirements/amd-linux-nightly.txt b/api/requirements/amd-linux-nightly.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
-torch==2.1.1
-torchvision==0.16.1
+torch==2.0.1
+torchvision==0.15.2
 https://download.onnxruntime.ai/onnxruntime_training-1.17.0.dev20230923001%2Brocm542-cp38-cp38-manylinux_2_28_x86_64.whl
diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt
@@ -1,4 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
-torch==2.1.1
-torchvision==0.16.1
+torch==2.0.1
+torchvision==0.15.2
 https://download.onnxruntime.ai/onnxruntime_training-1.16.3%2Brocm542-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl