• Pedro Cuenca's avatar
    Tests: upgrade PyTorch cuda to 11.7 to fix examples tests. (#1048) · a59f9990
    Pedro Cuenca authored
    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.
    a59f9990
push_tests.yml 2.93 KB