Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
89b8d4f5
Unverified
Commit
89b8d4f5
authored
Mar 08, 2021
by
Lysandre Debut
Committed by
GitHub
Mar 08, 2021
Browse files
Enable torch 1.8.0 on GPU CI (#10593)
* Enable torch 1.8.0 in GPU CI * Disable torch-scatter
parent
2a737bff
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
8 deletions
+4
-8
.github/workflows/self-push.yml
.github/workflows/self-push.yml
+2
-4
.github/workflows/self-scheduled.yml
.github/workflows/self-scheduled.yml
+2
-4
No files found.
.github/workflows/self-push.yml
View file @
89b8d4f5
...
...
@@ -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
:
|
...
...
.github/workflows/self-scheduled.yml
View file @
89b8d4f5
...
...
@@ -55,8 +55,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
pip list
- name
:
Are GPUs recognized by our DL frameworks
...
...
@@ -239,8 +238,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
pip install fairscale
pip install deepspeed
pip list
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment