torch version fix for python3.7

kubeflow · Jan 5, 2024 · b11ecc7 · b11ecc7
1 parent 7d7beff
commit b11ecc7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/sdk/python/kubeflow/storage_init_container/requirements.txt b/sdk/python/kubeflow/storage_init_container/requirements.txt
@@ -1,6 +1,6 @@
-torch>=1.8.2
-torchvision>=0.9.2
-torchaudio>=0.8.2
+torch>=1.8.1
+torchvision>=0.9.1
+torchaudio>=0.8.1
 einops==0.7.0
 transformers_stream_generator==0.0.4
 boto3==1.33.9

diff --git a/sdk/python/setup.py b/sdk/python/setup.py
@@ -31,9 +31,9 @@
     "retrying>=1.3.3",
     "boto3>=1.33.9",
     "transformers>=4.20.0",
-    "torch>=1.8.2",
-    "torchvision>=0.9.2",
-    "torchaudio>=0.8.2",
+    "torch>=1.8.1",
+    "torchvision>=0.9.1",
+    "torchaudio>=0.8.1",
     "einops>=0.7.0",
     "transformers_stream_generator>=0.0.4",
     "peft>=0.7.0",