From dc9aaa3848c886b4e15447ec986ff11867fb9d53 Mon Sep 17 00:00:00 2001 From: Lysandre Date: Fri, 5 Mar 2021 07:56:04 -0500 Subject: [PATCH] Pin torch to 1.7.1 in tests while we resolve issues --- .circleci/config.yml | 9 ++++++--- .github/workflows/self-push.yml | 6 ++++-- .github/workflows/self-scheduled.yml | 6 ++++-- 3 files changed, 14 insertions(+), 7 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index fe85b7aaa..b3e517587 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -79,7 +79,8 @@ jobs: - v0.4-{{ checksum "setup.py" }} - run: pip install --upgrade pip - run: pip install .[sklearn,tf-cpu,torch,testing,sentencepiece] - - run: pip install tapas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html + - run: pip install tapas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cpu.html + - run: pip install -U torch==1.7.1 - save_cache: key: v0.4-{{ checksum "setup.py" }} paths: @@ -106,7 +107,8 @@ jobs: - v0.4-{{ checksum "setup.py" }} - run: pip install --upgrade pip - run: pip install .[sklearn,torch,testing,sentencepiece] - - run: pip install tapas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html + - run: pip install tapas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cpu.html + - run: pip install -U torch==1.7.1 - save_cache: key: v0.4-torch-{{ checksum "setup.py" }} paths: @@ -185,7 +187,8 @@ jobs: - v0.4-{{ checksum "setup.py" }} - run: pip install --upgrade pip - run: pip install .[sklearn,torch,testing,sentencepiece] - - run: pip install tapas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.0+cpu.html + - run: pip install tapas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cpu.html + - run: pip install -U torch==1.7.1 - save_cache: key: v0.4-torch-{{ checksum "setup.py" }} paths: diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index f97e73c49..71f6f680d 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -54,7 +54,8 @@ 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.8.0+cu102.html + pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html + pip install -U torch==1.7.1 - name: Are GPUs recognized by our DL frameworks run: | @@ -201,7 +202,8 @@ 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.8.0+cu102.html + pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html + pip install -U torch==1.7.1 - name: Are GPUs recognized by our DL frameworks run: | diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index 6e6608fb9..33423f299 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -55,7 +55,8 @@ 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.8.0+cu102.html + 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 list - name: Are GPUs recognized by our DL frameworks @@ -238,7 +239,8 @@ 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.8.0+cu102.html + 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 fairscale pip install deepspeed pip list