update cu118 to use PT 2.1 (#2691)

* update cu118 to use PT 2.1 * update cu118 to use PT 2.1 * update cu118 to use PT 2.1 * reset yaml file
pytorch · Oct 10, 2023 · cd0679e · cd0679e
1 parent 6a42b5c
commit cd0679e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/requirements/torch_cu118_linux.txt b/requirements/torch_cu118_linux.txt
@@ -1,7 +1,7 @@
 #pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
 --extra-index-url https://download.pytorch.org/whl/cu118
 -r torch_common.txt
-torch==2.0.1+cu118; sys_platform == 'linux'
-torchvision==0.15.2+cu118; sys_platform == 'linux'
-torchtext==0.15.2; sys_platform == 'linux'
-torchaudio==2.0.2+cu118; sys_platform == 'linux'
+torch==2.1.0+cu118; sys_platform == 'linux'
+torchvision==0.16.0+cu118; sys_platform == 'linux'
+torchtext==0.16.0; sys_platform == 'linux'
+torchaudio==2.1.0+cu118; sys_platform == 'linux'
diff --git a/requirements/torch_cu118_windows.txt b/requirements/torch_cu118_windows.txt
@@ -1,7 +1,7 @@
 #pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
 --extra-index-url https://download.pytorch.org/whl/cu118
 -r torch_common.txt
-torch==2.0.1+cu118; sys_platform == 'linux'
-torchvision==0.15.2+cu118; sys_platform == 'linux'
-torchtext==0.15.2; sys_platform == 'linux'
-torchaudio==2.0.2+cu118; sys_platform == 'linux'
+torch==2.1.0+cu118; sys_platform == 'linux'
+torchvision==0.16.0+cu118; sys_platform == 'linux'
+torchtext==0.16.0; sys_platform == 'linux'
+torchaudio==2.1.0+cu118; sys_platform == 'linux'