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

[ci] Slow GPU tests run daily (#4465)

Browse Source
This commit is contained in:
Julien Chaumond
2020-05-25 17:28:02 -04:00
committed by GitHub
parent 3e3e552125
commit 365d452d4d
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -31,13 +31,12 @@ jobs:
- name: Install dependencies - name: Install dependencies
run: | run: |
source .env/bin/activate source .env/bin/activate
pip install .[sklearn,tf,torch,testing] pip install .[sklearn,torch,testing]
- 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(torch.cuda.is_available())"
python -c "import tensorflow as tf; print(tf.test.is_built_with_cuda(), tf.config.list_physical_devices('GPU'))"
- name: Run all tests on GPU - name: Run all tests on GPU
env: env: