From 44a93c981ff96f816e8b4e75e137d91b3ea03ae8 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Wed, 30 Sep 2020 12:53:20 +0200 Subject: [PATCH] Number of GPUs for multi-gpu (#7472) --- .github/workflows/self-push.yml | 3 ++- .github/workflows/self-scheduled.yml | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) 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: