Skip to content

GitLab

Unverified Commit 89b8d4f5 authored by Lysandre Debut's avatar Lysandre Debut Committed by GitHub
Browse files

Enable torch 1.8.0 on GPU CI (#10593) 

* Enable torch 1.8.0 in GPU CI

* Disable torch-scatter
parent 2a737bff
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 8 deletions
.github/workflows/self-push.yml
View file @ 89b8d4f5
...@@ -54,8 +54,7 @@ jobs: ...@@ -54,8 +54,7 @@ jobs:
pip install --upgrade pip pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece] pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install git+https://github.com/huggingface/datasets 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 pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install -U torch==1.7.1
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
run: | run: |
...@@ -202,8 +201,7 @@ jobs: ...@@ -202,8 +201,7 @@ jobs:
pip install --upgrade pip pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece] pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install git+https://github.com/huggingface/datasets 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 pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install -U torch==1.7.1
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
run: | run: |
......
.github/workflows/self-scheduled.yml
View file @ 89b8d4f5
...@@ -55,8 +55,7 @@ jobs: ...@@ -55,8 +55,7 @@ jobs:
pip install --upgrade pip pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece] pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install git+https://github.com/huggingface/datasets 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 pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install -U torch==1.7.1
pip list pip list
- name: Are GPUs recognized by our DL frameworks - name: Are GPUs recognized by our DL frameworks
...@@ -239,8 +238,7 @@ jobs: ...@@ -239,8 +238,7 @@ jobs:
pip install --upgrade pip pip install --upgrade pip
pip install .[torch,sklearn,testing,onnxruntime,sentencepiece] pip install .[torch,sklearn,testing,onnxruntime,sentencepiece]
pip install git+https://github.com/huggingface/datasets 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 pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cu102.html
pip install -U torch==1.7.1
pip install fairscale pip install fairscale
pip install deepspeed pip install deepspeed
pip list pip list
......
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