Skip to content

Commit 56bc60a

Browse files
committed
check if CUDA is available
1 parent 2126c0d commit 56bc60a

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

.github/workflows/test-linux-gpu.yml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,12 @@ jobs:
5555
-c "pytorch-${CHANNEL}" \
5656
-c nvidia "pytorch-${CHANNEL}"::pytorch[build="*${VERSION}*"] \
5757
"${CUDATOOLKIT}"
58+
59+
python3 -m torch.utils.collect_env
60+
python3 -c "import torch; exit(not torch.cuda.is_available())"
61+
5862
python3 setup.py develop
59-
python3 -m pip install pytest pytest-mock 'av<10'
63+
python3 -m pip install --progress-bar=off pytest pytest-mock 'av<10'
6064
6165
python3 -m pytest -vrA test/test_gpu_ci.py
6266

0 commit comments

Comments
 (0)