Skip to content

Commit ccc3f95

Browse files
agunapalmreso
andauthored
Upgrade to PyTorch 2.2.1 (#2959)
* Upgrade to PyTorch 2.2.1 * Update libtorch version --------- Co-authored-by: Matthias Reso <[email protected]>
1 parent e32e95a commit ccc3f95

8 files changed

+29
-29
lines changed

cpp/build.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ function install_kineto() {
7474
}
7575

7676
function install_libtorch() {
77-
TORCH_VERSION="2.2.0"
77+
TORCH_VERSION="2.2.1"
7878
if [ "$PLATFORM" = "Mac" ]; then
7979
if [ ! -d "$DEPS_DIR/libtorch" ]; then
8080
if [[ $(uname -m) == 'x86_64' ]]; then

requirements/torch_cu118_linux.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
22
--extra-index-url https://download.pytorch.org/whl/cu118
33
-r torch_common.txt
4-
torch==2.2.0+cu118; sys_platform == 'linux'
5-
torchvision==0.17.0+cu118; sys_platform == 'linux'
6-
torchtext==0.17.0; sys_platform == 'linux'
7-
torchaudio==2.2.0+cu118; sys_platform == 'linux'
4+
torch==2.2.1+cu118; sys_platform == 'linux'
5+
torchvision==0.17.1+cu118; sys_platform == 'linux'
6+
torchtext==0.17.1; sys_platform == 'linux'
7+
torchaudio==2.2.1+cu118; sys_platform == 'linux'

requirements/torch_cu118_windows.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
22
--extra-index-url https://download.pytorch.org/whl/cu118
33
-r torch_common.txt
4-
torch==2.2.0+cu118; sys_platform == 'linux'
5-
torchvision==0.17.0+cu118; sys_platform == 'linux'
6-
torchtext==0.17.0; sys_platform == 'linux'
7-
torchaudio==2.2.0+cu118; sys_platform == 'linux'
4+
torch==2.2.1+cu118; sys_platform == 'linux'
5+
torchvision==0.17.1+cu118; sys_platform == 'linux'
6+
torchtext==0.17.1; sys_platform == 'linux'
7+
torchaudio==2.2.1+cu118; sys_platform == 'linux'

requirements/torch_cu121_linux.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
22
--extra-index-url https://download.pytorch.org/whl/cu121
33
-r torch_common.txt
4-
torch==2.2.0+cu121; sys_platform == 'linux'
5-
torchvision==0.17.0+cu121; sys_platform == 'linux'
6-
torchtext==0.17.0; sys_platform == 'linux'
7-
torchaudio==2.2.0+cu121; sys_platform == 'linux'
4+
torch==2.2.1+cu121; sys_platform == 'linux'
5+
torchvision==0.17.1+cu121; sys_platform == 'linux'
6+
torchtext==0.17.1; sys_platform == 'linux'
7+
torchaudio==2.2.1+cu121; sys_platform == 'linux'

requirements/torch_cu121_windows.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121
22
--extra-index-url https://download.pytorch.org/whl/cu121
33
-r torch_common.txt
4-
torch==2.2.0+cu121; sys_platform == 'linux'
5-
torchvision==0.17.0+cu121; sys_platform == 'linux'
6-
torchtext==0.17.0; sys_platform == 'linux'
7-
torchaudio==2.2.0+cu121; sys_platform == 'linux'
4+
torch==2.2.1+cu121; sys_platform == 'linux'
5+
torchvision==0.17.1+cu121; sys_platform == 'linux'
6+
torchtext==0.17.1; sys_platform == 'linux'
7+
torchaudio==2.2.1+cu121; sys_platform == 'linux'

requirements/torch_darwin.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pip install torch torchvision torchaudio
22
--extra-index-url https://download.pytorch.org/whl/cpu
33
-r torch_common.txt
4-
torch==2.2.0; sys_platform == 'darwin'
5-
torchvision==0.17.0; sys_platform == 'darwin'
6-
torchtext==0.17.0; sys_platform == 'darwin'
7-
torchaudio==2.2.0; sys_platform == 'darwin'
4+
torch==2.2.1; sys_platform == 'darwin'
5+
torchvision==0.17.1; sys_platform == 'darwin'
6+
torchtext==0.17.1; sys_platform == 'darwin'
7+
torchaudio==2.2.1; sys_platform == 'darwin'

requirements/torch_linux.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu
22
--extra-index-url https://download.pytorch.org/whl/cpu
33
-r torch_common.txt
4-
torch==2.2.0+cpu; sys_platform == 'linux'
5-
torchvision==0.17.0+cpu; sys_platform == 'linux'
6-
torchtext==0.17.0; sys_platform == 'linux'
7-
torchaudio==2.2.0+cpu; sys_platform == 'linux'
4+
torch==2.2.1+cpu; sys_platform == 'linux'
5+
torchvision==0.17.1+cpu; sys_platform == 'linux'
6+
torchtext==0.17.1; sys_platform == 'linux'
7+
torchaudio==2.2.1+cpu; sys_platform == 'linux'

requirements/torch_windows.txt

+4-4
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#pip install torch torchvision torchaudio
22
--extra-index-url https://download.pytorch.org/whl/cpu
33
-r torch_common.txt
4-
torch==2.2.0; sys_platform == 'win32'
5-
torchvision==0.17.0; sys_platform == 'win32'
6-
torchtext==0.17.0; sys_platform == 'win32'
7-
torchaudio==2.2.0; sys_platform == 'win32'
4+
torch==2.2.1; sys_platform == 'win32'
5+
torchvision==0.17.1; sys_platform == 'win32'
6+
torchtext==0.17.1; sys_platform == 'win32'
7+
torchaudio==2.2.1; sys_platform == 'win32'

0 commit comments

Comments
 (0)