Skip to content

Commit 1a2d8bb

Browse files
committed
Update pip install to use whl/cpu
1 parent 036b1f0 commit 1a2d8bb

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/tests.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ jobs:
1717
matrix:
1818
os: [ubuntu-latest]
1919
python: ['3.10', '3.12']
20-
torch: [{base: '1.13.0', vision: '0.14.0'}, {base: '2.4.0', vision: '0.19.0'}]
20+
torch: [{base: '1.13.0', vision: '0.14.0'}, {base: '2.4.1', vision: '0.19.1'}]
2121
testmarker: ['-k "not test_models"', '-m base', '-m cfg', '-m torchscript', '-m features', '-m fxforward', '-m fxbackward']
2222
exclude:
2323
- python: '3.12'
@@ -46,7 +46,7 @@ jobs:
4646
sudo sed -i 's/azure\.//' /etc/apt/sources.list
4747
sudo apt update
4848
sudo apt install -y google-perftools
49-
pip install --no-cache-dir torch==${{ matrix.torch.base }}+cpu torchvision==${{ matrix.torch.vision }}+cpu -f https://download.pytorch.org/whl/torch_stable.html
49+
pip install --no-cache-dir torch==${{ matrix.torch.base }}+cpu torchvision==${{ matrix.torch.vision }}+cpu --index-url https://download.pytorch.org/whl/cpu
5050
- name: Install requirements
5151
run: |
5252
pip install -r requirements.txt

0 commit comments

Comments
 (0)