diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index 9f0f2d06a5..eabd6b9e1c 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -102,7 +102,8 @@ jobs: - name: Are GPUs recognized by our DL frameworks run: | source .env/bin/activate - python -c "import torch; print(torch.cuda.is_available())" + python -c "import torch; print('Cuda available:', torch.cuda.is_available())" + python -c "import torch; print('Number of GPUs available:', torch.cuda.device_count())" - name: Run all non-slow tests on GPU env: diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index d6c89a22ac..4a02f4a285 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -112,7 +112,8 @@ jobs: - name: Are GPUs recognized by our DL frameworks run: | source .env/bin/activate - python -c "import torch; print(torch.cuda.is_available())" + python -c "import torch; print('Cuda available:', torch.cuda.is_available())" + python -c "import torch; print('Number of GPUs available:', torch.cuda.device_count())" - name: Run all tests on GPU env: