From 35e94c68df40c7e951512cdf21c0b0d84b3591ad Mon Sep 17 00:00:00 2001 From: Lysandre Date: Wed, 30 Sep 2020 12:29:26 +0200 Subject: [PATCH] Number of GPUs --- .github/workflows/self-push.yml | 3 ++- .github/workflows/self-scheduled.yml | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index 5ac040b920..9f0f2d06a5 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -51,7 +51,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 e70be8cd09..d6c89a22ac 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -48,7 +48,9 @@ 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: