Merge pull request #10 from sony/torch_cpu

torch cpu
sony · Mar 31, 2024 · 8b6a240 · 8b6a240
2 parents 1b2b5bf + 27e277b
commit 8b6a240
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
@@ -72,7 +72,10 @@ jobs:
                       torchvision==${{matrix.torchvision_ver}} \
                       onnxruntime==${{matrix.ort_ver}} \
                       onnxruntime_extensions==${{matrix.ort_ext_ver}} \
-                      onnx==${{matrix.onnx_ver}}
+                      onnx==${{matrix.onnx_ver}} \
+                      --index-url https://download.pytorch.org/whl/cpu \
+                      --extra-index-url https://pypi.org/simple
+
           pip install -r requirements_test.txt
           pip list
       - name: Run pytest