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

Enable torch 1.8.0 on GPU CI (#10593)

Browse Source 
* Enable torch 1.8.0 in GPU CI

* Disable torch-scatter
This commit is contained in:
Lysandre Debut
2021-03-08 07:11:43 -05:00
committed by GitHub
parent 2a737bffef
commit 89b8d4f568
2 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -54,8 +54,7 @@ jobs:
pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install git+https://github.com/huggingface/datasets
pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install -U torch==1.7.1
# pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
- name: Are GPUs recognized by our DL frameworks
run: |
@@ -202,8 +201,7 @@ jobs:
pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install git+https://github.com/huggingface/datasets
pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html
pip install -U torch==1.7.1
# pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
- name: Are GPUs recognized by our DL frameworks
run: |