From 0aab1301aa1f991cd52190cd88088dc6e69befce Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Thu, 21 Dec 2023 18:57:00 -0600 Subject: [PATCH] fix(api): revert torch to 2.0.1 for ROCm due to MIopen errors (#444) --- api/requirements/amd-linux-nightly.txt | 4 ++-- api/requirements/amd-linux.txt | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/api/requirements/amd-linux-nightly.txt b/api/requirements/amd-linux-nightly.txt index f9a237553..892c92098 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.4.2 -torch==2.1.1 -torchvision==0.16.1 +torch==2.0.1 +torchvision==0.15.2 https://download.onnxruntime.ai/onnxruntime_training-1.17.0.dev20230923001%2Brocm542-cp38-cp38-manylinux_2_28_x86_64.whl \ No newline at end of file diff --git a/api/requirements/amd-linux.txt b/api/requirements/amd-linux.txt index ba7186c57..961ff9f75 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.4.2 -torch==2.1.1 -torchvision==0.16.1 +torch==2.0.1 +torchvision==0.15.2 https://download.onnxruntime.ai/onnxruntime_training-1.16.3%2Brocm542-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl \ No newline at end of file