From 89b8d4f568d7939625237f8797bb737877b90400 Mon Sep 17 00:00:00 2001 From: Lysandre Debut Date: Mon, 8 Mar 2021 07:11:43 -0500 Subject: [PATCH] Enable torch 1.8.0 on GPU CI (#10593) * Enable torch 1.8.0 in GPU CI * Disable torch-scatter --- .github/workflows/self-push.yml | 6 ++---- .github/workflows/self-scheduled.yml | 6 ++---- 2 files changed, 4 insertions(+), 8 deletions(-) diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index 71f6f680d8..0a26aab136 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -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: | diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 33423f299c..186a40dbe3 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -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