From dc9f24544291b25b44c9e87239a0ef4355396a4c Mon Sep 17 00:00:00 2001 From: Lysandre Date: Wed, 16 Dec 2020 13:48:52 -0500 Subject: [PATCH] Torch scatter with torch 1.7.0 --- .github/workflows/self-push.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/self-push.yml b/.github/workflows/self-push.yml index 68e0dc7310..c81f64c015 100644 --- a/.github/workflows/self-push.yml +++ b/.github/workflows/self-push.yml @@ -50,7 +50,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-$(python -c "import torch; print(''.join(torch.__version__)")+cu102.html + pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html - name: Are GPUs recognized by our DL frameworks run: | @@ -188,7 +188,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-$(python -c "import torch; print(''.join(torch.__version__)")+cu102.html + pip install pandas torch-scatter -f https://pytorch-geometric.com/whl/torch-1.7.0+cu102.html - name: Are GPUs recognized by our DL frameworks run: |