diff --git a/.github/workflows/python-test.yml b/.github/workflows/python-test.yml index 3a42ee9..fe54d29 100644 --- a/.github/workflows/python-test.yml +++ b/.github/workflows/python-test.yml @@ -28,7 +28,7 @@ jobs: python -m pip install --upgrade pip python -m pip install pytest python -m pip install wheel - python -m pip install torch==2.4.0+cpu torchvision==0.19.0+cpu + python -m pip install torch==2.4.0 torchvision==0.19.0 --index-url https://download.pytorch.org/whl/cpu if [ -f requirements.txt ]; then pip install -r requirements.txt; fi - name: Test with pytest run: |