Install particular version of the PyTorch package.

ybubnov · Apr 19, 2024 · a30870a · a30870a
1 parent ef28a23
commit a30870a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/.github/workflows/cmake.yaml b/.github/workflows/cmake.yaml
@@ -32,8 +32,9 @@ jobs:
 
     - name: Install PyTorch (CPU)
       id: install_torch_cpu
-      run: |
-        pip3 install torch+cpu --index-url --index-url https://download.pytorch.org/whl/cpu
+      run: >
+        pip3 install 'torch>=2.0.0+cpu'
+        --index-url --index-url https://download.pytorch.org/whl/cpu
 
     - name: Find PyTorch CMake path
       id: find_libtorch