Skip to content

GitLab

Commit 35e94c68 authored by Lysandre's avatar Lysandre
Browse files

Number of GPUs

parent 056723ad
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
.github/workflows/self-push.yml
View file @ 35e94c68
......@@ -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:
......
.github/workflows/self-scheduled.yml
View file @ 35e94c68
......@@ -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:
......
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