From 3a1724652e2d2d6ebf070f39664393fb2ffa3a15 Mon Sep 17 00:00:00 2001 From: Lincoln Stein Date: Mon, 16 Jan 2023 14:19:27 -0500 Subject: [PATCH] upgrade requirements to CUDA 11.7, torch 1.13 (#2331) * upgrade requirements to CUDA 11.7, torch 1.13 * fix ROCm version number Co-authored-by: Lincoln Stein --- environments-and-requirements/requirements-lin-amd.txt | 2 +- environments-and-requirements/requirements-lin-cuda.txt | 2 +- environments-and-requirements/requirements-win-colab-cuda.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/environments-and-requirements/requirements-lin-amd.txt b/environments-and-requirements/requirements-lin-amd.txt index 0aa6092b30..0e707be483 100644 --- a/environments-and-requirements/requirements-lin-amd.txt +++ b/environments-and-requirements/requirements-lin-amd.txt @@ -1,6 +1,6 @@ -r environments-and-requirements/requirements-base.txt # Get hardware-appropriate torch/torchvision ---extra-index-url https://download.pytorch.org/whl/rocm5.1.1 --trusted-host https://download.pytorch.org +--extra-index-url https://download.pytorch.org/whl/rocm5.2 --trusted-host https://download.pytorch.org torch>=1.13.1 torchvision>=0.14.1 -e . diff --git a/environments-and-requirements/requirements-lin-cuda.txt b/environments-and-requirements/requirements-lin-cuda.txt index d7ea4b4542..1f1ebcf947 100644 --- a/environments-and-requirements/requirements-lin-cuda.txt +++ b/environments-and-requirements/requirements-lin-cuda.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/cu116 --trusted-host https://download.pytorch.org +--trusted-host https://download.pytorch.org -r environments-and-requirements/requirements-base.txt torch>=1.13.1 torchvision>=0.14.1 diff --git a/environments-and-requirements/requirements-win-colab-cuda.txt b/environments-and-requirements/requirements-win-colab-cuda.txt index 01b25144b6..7fc55c538a 100644 --- a/environments-and-requirements/requirements-win-colab-cuda.txt +++ b/environments-and-requirements/requirements-win-colab-cuda.txt @@ -1,6 +1,6 @@ -r environments-and-requirements/requirements-base.txt # Get hardware-appropriate torch/torchvision ---extra-index-url https://download.pytorch.org/whl/cu116 --trusted-host https://download.pytorch.org +--extra-index-url https://download.pytorch.org/whl/cu117 --trusted-host https://download.pytorch.org torch==1.13.1 torchvision==0.14.1 -e .