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