Install torch CPU

(cherry picked from commit dcc5fcd)
apaniukov · Apr 18, 2024 · 4891f99 · 4891f99
1 parent 6d6ec53
commit 4891f99
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/.github/workflows/linux.yml b/.github/workflows/linux.yml
@@ -329,6 +329,8 @@ jobs:
             wheel_name=$(find . -name 'openvino_tokenizers*.whl')
             python3 -m pip install $wheel_name[dev]
           popd
+        env:
+          PIP_EXTRA_INDEX_URL: https://download.pytorch.org/whl/cpu
 
       - name: Tokenizers regression tests (using openvino python modules)
         if: needs.openvino_download.outputs.status == 'success'

diff --git a/.github/workflows/mac.yml b/.github/workflows/mac.yml
@@ -319,6 +319,8 @@ jobs:
             wheel_name=$(find . -name 'openvino_tokenizers*.whl')
             python3 -m pip install $wheel_name[dev]
           popd
+        env:
+          PIP_EXTRA_INDEX_URL: https://download.pytorch.org/whl/cpu
 
       - name: Tokenizers regression tests (using openvino python modules)
         if: needs.openvino_download.outputs.status == 'success'

diff --git a/.github/workflows/windows.yml b/.github/workflows/windows.yml
@@ -323,6 +323,8 @@ jobs:
           # Find and install wheel
           $ovCoreWheelPath=Get-ChildItem -Path "${{ env.INSTALL_DIR }}\\ov_tokenizers" -Filter openvino_tokenizers*.whl | % { $_.FullName }
           python3 -m pip install "$ovCoreWheelPath[all]"
+        env:
+          PIP_EXTRA_INDEX_URL: https://download.pytorch.org/whl/cpu
 
       - name: Tokenizers regression tests (using openvino python modules)
         if: needs.openvino_download.outputs.status == 'success'