Move +cpu from matrix to pip command

.github/workflows/ci-sharktank.yml

@@ -27,16 +27,16 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.11", "3.12"]
-        torch-version: ["2.3.0+cpu", "2.4.1+cpu", "2.5.1+cpu"]
+        torch-version: ["2.3.0", "2.4.1", "2.5.1"]
         os: [ubuntu-24.04]
         include:
           - os: windows-2022
             python-version: "3.11"
-            torch-version: "2.3.0+cpu"
+            torch-version: "2.3.0"
         exclude:
           - python-version: "3.12"
              # `torch.compile` requires torch>=2.4.0 for Python 3.12+
-            torch-version: "2.3.0+cpu"
+            torch-version: "2.3.0"
       fail-fast: false
     runs-on: ${{matrix.os}}
     defaults:
@@ -66,7 +66,7 @@ jobs:
           # Note: We install in three steps in order to satisfy requirements
           # from non default locations first. Installing the PyTorch CPU
           # wheels saves multiple minutes and a lot of bandwidth on runner setup.
-          pip install --no-compile --pre --index-url https://download.pytorch.org/whl/test/cpu torch==${{matrix.torch-version}}
+          pip install --no-compile --pre --index-url https://download.pytorch.org/whl/test/cpu torch==${{matrix.torch-version}}+cpu
           pip install --no-compile -r requirements.txt -r sharktank/requirements-tests.txt -e sharktank/
 
           # Install nightly IREE packages.