From 30fd55260a21a07c5473636a3f3b18dbcf1b219a Mon Sep 17 00:00:00 2001 From: agunapal Date: Thu, 22 Feb 2024 18:15:21 +0000 Subject: [PATCH 1/2] Upgrade to PyTorch 2.2.1 --- requirements/torch_cu118_linux.txt | 8 ++++---- requirements/torch_cu118_windows.txt | 8 ++++---- requirements/torch_cu121_linux.txt | 8 ++++---- requirements/torch_cu121_windows.txt | 8 ++++---- requirements/torch_darwin.txt | 8 ++++---- requirements/torch_linux.txt | 8 ++++---- requirements/torch_windows.txt | 8 ++++---- 7 files changed, 28 insertions(+), 28 deletions(-) diff --git a/requirements/torch_cu118_linux.txt b/requirements/torch_cu118_linux.txt index 61348115a6..69e1c6dbc7 100644 --- 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.2.0+cu118; sys_platform == 'linux' -torchvision==0.17.0+cu118; sys_platform == 'linux' -torchtext==0.17.0; sys_platform == 'linux' -torchaudio==2.2.0+cu118; sys_platform == 'linux' +torch==2.2.1+cu118; sys_platform == 'linux' +torchvision==0.17.1+cu118; sys_platform == 'linux' +torchtext==0.17.1; sys_platform == 'linux' +torchaudio==2.2.1+cu118; sys_platform == 'linux' diff --git a/requirements/torch_cu118_windows.txt b/requirements/torch_cu118_windows.txt index 61348115a6..69e1c6dbc7 100644 --- 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.2.0+cu118; sys_platform == 'linux' -torchvision==0.17.0+cu118; sys_platform == 'linux' -torchtext==0.17.0; sys_platform == 'linux' -torchaudio==2.2.0+cu118; sys_platform == 'linux' +torch==2.2.1+cu118; sys_platform == 'linux' +torchvision==0.17.1+cu118; sys_platform == 'linux' +torchtext==0.17.1; sys_platform == 'linux' +torchaudio==2.2.1+cu118; sys_platform == 'linux' diff --git a/requirements/torch_cu121_linux.txt b/requirements/torch_cu121_linux.txt index f594c1c5cb..7ab3acdcf4 100644 --- a/requirements/torch_cu121_linux.txt +++ b/requirements/torch_cu121_linux.txt @@ -1,7 +1,7 @@ #pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121 --extra-index-url https://download.pytorch.org/whl/cu121 -r torch_common.txt -torch==2.2.0+cu121; sys_platform == 'linux' -torchvision==0.17.0+cu121; sys_platform == 'linux' -torchtext==0.17.0; sys_platform == 'linux' -torchaudio==2.2.0+cu121; sys_platform == 'linux' +torch==2.2.1+cu121; sys_platform == 'linux' +torchvision==0.17.1+cu121; sys_platform == 'linux' +torchtext==0.17.1; sys_platform == 'linux' +torchaudio==2.2.1+cu121; sys_platform == 'linux' diff --git a/requirements/torch_cu121_windows.txt b/requirements/torch_cu121_windows.txt index f594c1c5cb..7ab3acdcf4 100644 --- a/requirements/torch_cu121_windows.txt +++ b/requirements/torch_cu121_windows.txt @@ -1,7 +1,7 @@ #pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu121 --extra-index-url https://download.pytorch.org/whl/cu121 -r torch_common.txt -torch==2.2.0+cu121; sys_platform == 'linux' -torchvision==0.17.0+cu121; sys_platform == 'linux' -torchtext==0.17.0; sys_platform == 'linux' -torchaudio==2.2.0+cu121; sys_platform == 'linux' +torch==2.2.1+cu121; sys_platform == 'linux' +torchvision==0.17.1+cu121; sys_platform == 'linux' +torchtext==0.17.1; sys_platform == 'linux' +torchaudio==2.2.1+cu121; sys_platform == 'linux' diff --git a/requirements/torch_darwin.txt b/requirements/torch_darwin.txt index 40fb6aa249..d56352b8a2 100644 --- a/requirements/torch_darwin.txt +++ b/requirements/torch_darwin.txt @@ -1,7 +1,7 @@ #pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu -r torch_common.txt -torch==2.2.0; sys_platform == 'darwin' -torchvision==0.17.0; sys_platform == 'darwin' -torchtext==0.17.0; sys_platform == 'darwin' -torchaudio==2.2.0; sys_platform == 'darwin' +torch==2.2.1; sys_platform == 'darwin' +torchvision==0.17.1; sys_platform == 'darwin' +torchtext==0.17.1; sys_platform == 'darwin' +torchaudio==2.2.1; sys_platform == 'darwin' diff --git a/requirements/torch_linux.txt b/requirements/torch_linux.txt index 31c8b35a2d..460e3c4d5b 100644 --- a/requirements/torch_linux.txt +++ b/requirements/torch_linux.txt @@ -1,7 +1,7 @@ #pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu --extra-index-url https://download.pytorch.org/whl/cpu -r torch_common.txt -torch==2.2.0+cpu; sys_platform == 'linux' -torchvision==0.17.0+cpu; sys_platform == 'linux' -torchtext==0.17.0; sys_platform == 'linux' -torchaudio==2.2.0+cpu; sys_platform == 'linux' +torch==2.2.1+cpu; sys_platform == 'linux' +torchvision==0.17.1+cpu; sys_platform == 'linux' +torchtext==0.17.1; sys_platform == 'linux' +torchaudio==2.2.1+cpu; sys_platform == 'linux' diff --git a/requirements/torch_windows.txt b/requirements/torch_windows.txt index bb4fac9b4d..937d616b77 100644 --- a/requirements/torch_windows.txt +++ b/requirements/torch_windows.txt @@ -1,7 +1,7 @@ #pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cpu -r torch_common.txt -torch==2.2.0; sys_platform == 'win32' -torchvision==0.17.0; sys_platform == 'win32' -torchtext==0.17.0; sys_platform == 'win32' -torchaudio==2.2.0; sys_platform == 'win32' +torch==2.2.1; sys_platform == 'win32' +torchvision==0.17.1; sys_platform == 'win32' +torchtext==0.17.1; sys_platform == 'win32' +torchaudio==2.2.1; sys_platform == 'win32' From 381040e870f35d89507d3ef31a30cbf289be965e Mon Sep 17 00:00:00 2001 From: Matthias Reso <13337103+mreso@users.noreply.github.com> Date: Thu, 22 Feb 2024 18:39:54 +0000 Subject: [PATCH 2/2] Update libtorch version --- cpp/build.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/build.sh b/cpp/build.sh index 57fad8c8e8..5beb2b79b5 100755 --- a/cpp/build.sh +++ b/cpp/build.sh @@ -74,7 +74,7 @@ function install_kineto() { } function install_libtorch() { - TORCH_VERSION="2.2.0" + TORCH_VERSION="2.2.1" if [ "$PLATFORM" = "Mac" ]; then if [ ! -d "$DEPS_DIR/libtorch" ]; then if [[ $(uname -m) == 'x86_64' ]]; then