{{ message }}
Repository navigation
Commit a59f999
authored
Tests: upgrade PyTorch cuda to 11.7 to fix examples tests. (huggingface#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
1 file changed
Lines changed: 2 additions & 2 deletions
0 commit comments