chore(example): upgrade pytorch version for examples (#2548)

star-whale · Jul 24, 2023 · e3eeb75 · e3eeb75
1 parent 6041841
commit e3eeb75
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 18 deletions.
diff --git a/example/runtime/pytorch-cn-mirror/requirements.txt b/example/runtime/pytorch-cn-mirror/requirements.txt
@@ -15,9 +15,10 @@ numpy==1.23.2; python_version>='3.8'
 scikit-learn==1.1.1; python_version>='3.8'
 scipy==1.8.1; python_version>='3.8'
 importlib-metadata>=4.0.0, <=4.2.0;python_version < '3.8'
-torch==1.12.1
-torchaudio==0.12.1
-torchdata==0.4.1
-torchtext==0.13.1
-torchvision==0.13.1
-gradio
+torch==2.0.1; python_version>='3.8'
+torch==1.13.1; python_version=='3.7'
+torchaudio
+torchdata
+torchtext
+torchvision
+gradio
diff --git a/example/runtime/pytorch-e2e/requirements.txt b/example/runtime/pytorch-e2e/requirements.txt
@@ -11,8 +11,9 @@ numpy==1.23.2; python_version>='3.8'
 scikit-learn==1.1.1; python_version>='3.8'
 scipy==1.8.1; python_version>='3.8'
 importlib-metadata>=4.0.0, <=4.2.0;python_version < '3.8'
-torch==1.12.1
-torchaudio==0.12.1
-torchdata==0.4.1
-torchtext==0.13.1
-torchvision==0.13.1
+torch==2.0.1; python_version>='3.8'
+torch==1.13.1; python_version=='3.7'
+torchaudio
+torchdata
+torchtext
+torchvision
diff --git a/example/runtime/pytorch/requirements.txt b/example/runtime/pytorch/requirements.txt
@@ -1,4 +1,3 @@
-starwhale>=0.3.6
 # for pfp
 pycocotools
 # for ucf101
@@ -11,9 +10,10 @@ numpy==1.23.2; python_version>='3.8'
 scikit-learn==1.1.1; python_version>='3.8'
 scipy==1.8.1; python_version>='3.8'
 importlib-metadata>=4.0.0, <=4.2.0;python_version < '3.8'
-torch==1.12.1
-torchaudio==0.12.1
-torchdata==0.4.1
-torchtext==0.13.1
-torchvision==0.13.1
-gradio
+torch==2.0.1; python_version>='3.8'
+torch==1.13.1; python_version=='3.7'
+torchaudio
+torchdata
+torchtext
+torchvision
+gradio