From 89ebbb8536a5365ea9152e177285515d5db1e694 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 9 Apr 2023 17:19:40 -0500 Subject: [PATCH] feat(api): update requirements for torch 2.0 --- api/requirements/amd-linux-nightly.txt | 6 +++--- api/requirements/amd-linux.txt | 6 +++--- api/requirements/amd-windows-nightly.txt | 4 ++-- api/requirements/amd-windows.txt | 4 ++-- api/requirements/cpu-nightly.txt | 4 ++-- api/requirements/cpu.txt | 4 ++-- api/requirements/nvidia-nightly.txt | 4 ++-- api/requirements/nvidia.txt | 4 ++-- 8 files changed, 18 insertions(+), 18 deletions(-) diff --git a/api/requirements/amd-linux-nightly.txt b/api/requirements/amd-linux-nightly.txt index ba2c48df..aae793aa 100644 --- a/api/requirements/amd-linux-nightly.txt +++ b/api/requirements/amd-linux-nightly.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/rocm5.2 -torch==1.13.1 -torchvision==0.14.1 +--extra-index-url https://download.pytorch.org/whl/rocm5.4.2 +torch==2.0.0 +torchvision==0.15.1 https://download.onnxruntime.ai/onnxruntime_training-1.15.0.dev20230326001%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl \ No newline at end of file diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt index 1405f187..c44cd431 100644 --- a/api/requirements/amd-linux.txt +++ b/api/requirements/amd-linux.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/rocm5.2 -torch==1.13.1 -torchvision==0.14.1 +--extra-index-url https://download.pytorch.org/whl/rocm5.4.2 +torch==2.0.0 +torchvision==0.15.1 https://download.onnxruntime.ai/onnxruntime_training-1.14.1%2Brocm54-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl \ No newline at end of file diff --git a/api/requirements/amd-windows-nightly.txt b/api/requirements/amd-windows-nightly.txt index 99e17d6b..0e5853dc 100644 --- a/api/requirements/amd-windows-nightly.txt +++ b/api/requirements/amd-windows-nightly.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cpu -torch==1.13.1 -torchvision==0.14.1 +torch==2.0.0 +torchvision==0.15.1 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ ort-nightly-directml \ No newline at end of file diff --git a/api/requirements/amd-windows.txt b/api/requirements/amd-windows.txt index 694aaa40..a6842263 100644 --- a/api/requirements/amd-windows.txt +++ b/api/requirements/amd-windows.txt @@ -1,4 +1,4 @@ --extra-index-url https://download.pytorch.org/whl/cpu -torch==1.13.1 -torchvision==0.14.1 +torch==2.0.0 +torchvision==0.15.1 onnxruntime-directml>=1.13 \ No newline at end of file diff --git a/api/requirements/cpu-nightly.txt b/api/requirements/cpu-nightly.txt index 0b9d115a..408f5706 100644 --- a/api/requirements/cpu-nightly.txt +++ b/api/requirements/cpu-nightly.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cpu -torch==1.13.1 -torchvision==0.14.1 +torch==2.0.0 +torchvision==0.15.1 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ ort-nightly \ No newline at end of file diff --git a/api/requirements/cpu.txt b/api/requirements/cpu.txt index a55fbd60..c5cb43d5 100644 --- a/api/requirements/cpu.txt +++ b/api/requirements/cpu.txt @@ -1,4 +1,4 @@ --extra-index-url https://download.pytorch.org/whl/cpu -torch==1.13.1 -torchvision==0.14.1 +torch==2.0.0 +torchvision==0.15.1 onnxruntime>=1.13 diff --git a/api/requirements/nvidia-nightly.txt b/api/requirements/nvidia-nightly.txt index bd753ac4..6070337f 100644 --- a/api/requirements/nvidia-nightly.txt +++ b/api/requirements/nvidia-nightly.txt @@ -1,5 +1,5 @@ --extra-index-url https://download.pytorch.org/whl/cu117 -torch==1.13.1 -torchvision==0.14.1 +torch==2.0.0 +torchvision==0.15.1 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/ORT-Nightly/pypi/simple/ ort-nightly-gpu \ No newline at end of file diff --git a/api/requirements/nvidia.txt b/api/requirements/nvidia.txt index d687b589..bba2ce6c 100644 --- a/api/requirements/nvidia.txt +++ b/api/requirements/nvidia.txt @@ -1,4 +1,4 @@ --extra-index-url https://download.pytorch.org/whl/cu117 -torch==1.13.1 -torchvision==0.14.1 +torch==2.0.0 +torchvision==0.15.1 onnxruntime-gpu>=1.13