From e1ed4bd6e5a4bda6365ae0eeae939212aa768889 Mon Sep 17 00:00:00 2001 From: Casper Date: Wed, 14 Feb 2024 20:04:01 +0100 Subject: [PATCH] Remove +rocm from pip install --- .github/workflows/build.yaml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 02ed8c3..2aee262 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -209,9 +209,9 @@ jobs: python -m pip install --upgrade build setuptools wheel if [[ "${{ matrix.rocm }}" == "5.7.1" ]]; then - python -m pip install torch==2.2.0+rocm57 --index-url https://download.pytorch.org/whl/rocm5.7 + python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.7 elif [[ "${{ matrix.rocm }}" == "5.6.1" ]]; then - python -m pip install torch==2.2.0+rocm56 --index-url https://download.pytorch.org/whl/rocm5.6 + python -m pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/rocm5.6 else echo Unknown rocm version for python install exit 1