We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2126c0d commit 56bc60aCopy full SHA for 56bc60a
.github/workflows/test-linux-gpu.yml
@@ -55,8 +55,12 @@ jobs:
55
-c "pytorch-${CHANNEL}" \
56
-c nvidia "pytorch-${CHANNEL}"::pytorch[build="*${VERSION}*"] \
57
"${CUDATOOLKIT}"
58
+
59
+ python3 -m torch.utils.collect_env
60
+ python3 -c "import torch; exit(not torch.cuda.is_available())"
61
62
python3 setup.py develop
- python3 -m pip install pytest pytest-mock 'av<10'
63
+ python3 -m pip install --progress-bar=off pytest pytest-mock 'av<10'
64
65
python3 -m pytest -vrA test/test_gpu_ci.py
66
0 commit comments