try torch==2.2.0

.github/workflows/pythonapp.yml

@@ -28,6 +28,7 @@ jobs:
         python-version: ['3.8', '3.9', '3.10', '3.11']
     env:
       TVM_VERSION_TAG: v0.15.0
+      PYTORCH_VERSION: 2.2.0
 
     steps:
     - uses: actions/checkout@v4
@@ -44,12 +45,12 @@ jobs:
         # The GitHub editor is 127 chars wide
         flake8 . --count  --max-complexity=10 --max-line-length=127 --statistics
     # PyTorch CPU for Linux has different pip syntax wrt Win and Mac.
-    - name: Install torch-2.0.0+cpu if linux
+    - name: Install torch-${{ env.PYTORCH_VERSION }}+cpu if linux
       if: ${{ startsWith(matrix.os, 'ubuntu') }}
-      run: python -m pip install torch==2.0.0 --extra-index-url https://download.pytorch.org/whl/cpu
-    - name: Install torch-2.0.0+cpu if not linux
+      run: python -m pip install torch==${{ env.PYTORCH_VERSION }} --extra-index-url https://download.pytorch.org/whl/cpu
+    - name: Install torch-${{ env.PYTORCH_VERSION }}+cpu if not linux
       if:  ${{ !startsWith(matrix.os, 'ubuntu') }}
-      run: python -m pip install torch==2.0.0
+      run: python -m pip install torch==${{ env.PYTORCH_VERSION }}
     - name: Install basic dependencies
       run: |
         python -m pip install -e .[tests] -f https://download.pytorch.org/whl/torch_stable.html