update torch/ipex version in CI

intel · Jan 12, 2024 · c6f43f8 · c6f43f8
1 parent d74ab85
commit c6f43f8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml
@@ -46,7 +46,8 @@ jobs:
         if: ${{ env.BACKEND == 'XPU'}}
         run: |
           python3 -m pip install lit
-          python3 -m pip install intel_extension_for_pytorch
+          python3 -m pip install torch==2.1.0a0+cxx11.abi intel_extension_for_pytorch==2.1.10+xpu -f https://developer.intel.com/ipex-whl-stable-xpu
+
           cd python
           LIT_TEST_DIR="build/$(ls build | grep -i cmake)/test"
           if [ ! -d "${LIT_TEST_DIR}" ]; then