From a59f9990fc1251ada20e86dfa615b04360bd013f Mon Sep 17 00:00:00 2001 From: Pedro Cuenca Date: Sat, 29 Oct 2022 20:27:00 +0200 Subject: [PATCH] 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. --- .github/workflows/push_tests.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/push_tests.yml b/.github/workflows/push_tests.yml index dfd83aa9af..f1fc5484a2 100644 --- a/.github/workflows/push_tests.yml +++ b/.github/workflows/push_tests.yml @@ -34,7 +34,7 @@ jobs: run: | python -m pip install --upgrade pip python -m pip uninstall -y torch torchvision torchtext - python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu116 + python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu117 python -m pip install -e .[quality,test] python -m pip install git+https://github.com/huggingface/accelerate @@ -80,7 +80,7 @@ jobs: run: | python -m pip install --upgrade pip python -m pip uninstall -y torch torchvision torchtext - python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu116 + python -m pip install torch --extra-index-url https://download.pytorch.org/whl/cu117 python -m pip install -e .[quality,test,training] python -m pip install git+https://github.com/huggingface/accelerate