Skip to content

Commit a59f999

Browse files
authored
Tests: upgrade PyTorch cuda to 11.7 to fix examples tests. (#1048)
Tests: upgrade PyTorch cuda to 11.7. Otherwise the cuda versions of torch and torchvision mismatch, and examples tests fail. We were requesting cuda 11.6 for PyTorch, and the default torchvision (via setup.py). Another option would be to include torchvision in the same pip install line as torch.
1 parent 1fc2088 commit a59f999

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/workflows/push_tests.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ jobs:
3434
run: |
3535
python -m pip install --upgrade pip
3636
python -m pip uninstall -y torch torchvision torchtext
37-
python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu116
37+
python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu117
3838
python -m pip install -e .[quality,test]
3939
python -m pip install git+https://github.com/huggingface/accelerate
4040
@@ -80,7 +80,7 @@ jobs:
8080
run: |
8181
python -m pip install --upgrade pip
8282
python -m pip uninstall -y torch torchvision torchtext
83-
python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu116
83+
python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu117
8484
python -m pip install -e .[quality,test,training]
8585
python -m pip install git+https://github.com/huggingface/accelerate
8686

0 commit comments

Comments
 (0)