diff --git a/.github/workflows/test-linux-gpu.yml b/.github/workflows/test-linux-gpu.yml index a4d938f23ed..37a4256a19d 100644 --- a/.github/workflows/test-linux-gpu.yml +++ b/.github/workflows/test-linux-gpu.yml @@ -51,8 +51,12 @@ jobs: conda install \ --yes \ -c "pytorch-${CHANNEL}" \ - -c nvidia "pytorch-${CHANNEL}"::pytorch[build="*${VERSION}*"] \ + -c nvidia \ + pytorch \ "${CUDATOOLKIT}" + + python3 -c "import torch; exit(not torch.cuda.is_available())" + python3 setup.py develop python3 -m pip install pytest pytest-mock 'av<10'