Skip to content

GitLab

Unverified Commit 44a93c98 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Number of GPUs for multi-gpu (#7472)

parent 886ef35c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
.github/workflows/self-push.yml
View file @ 44a93c98
......@@ -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:
......
.github/workflows/self-scheduled.yml
View file @ 44a93c98
......@@ -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:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment