SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Number of GPUs for multi-gpu (#7472)

Browse Source
This commit is contained in:
Lysandre Debut
2020-09-30 12:53:20 +02:00
committed by GitHub
parent 886ef35ce6
commit 44a93c981f
2 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.github/workflows/self-push.yml vendored
View File

@@ -102,7 +102,8 @@ jobs:
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
run: | run: |
source .env/bin/activate 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 - name: Run all non-slow tests on GPU
env: env:

3
.github/workflows/self-scheduled.yml vendored
View File

@@ -112,7 +112,8 @@ jobs:
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
run: | run: |
source .env/bin/activate 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 - name: Run all tests on GPU
env: env: