Unverified Commit 381d28ca authored by Benjamin Lefaudeux's avatar Benjamin Lefaudeux Committed by GitHub
Browse files

[fix] CircleCI vs pip hotfix (#267)

* keep two torch 1.7 profiles to save cuda 10.1 testing
parent c386e937
...@@ -59,7 +59,7 @@ install_dep_16: &install_dep_16 ...@@ -59,7 +59,7 @@ install_dep_16: &install_dep_16
python -c 'import torch; print("Torch version:", torch.__version__)' python -c 'import torch; print("Torch version:", torch.__version__)'
python -m torch.utils.collect_env python -m torch.utils.collect_env
install_dep_17: &install_dep_17 install_dep_17_cpu: &install_dep_17_cpu
- run: - run:
name: Install Dependencies name: Install Dependencies
command: | command: |
...@@ -70,6 +70,20 @@ install_dep_17: &install_dep_17 ...@@ -70,6 +70,20 @@ install_dep_17: &install_dep_17
python -c 'import torch; print("Torch version:", torch.__version__)' python -c 'import torch; print("Torch version:", torch.__version__)'
python -m torch.utils.collect_env python -m torch.utils.collect_env
install_dep_17_gpu: &install_dep_17_gpu
# FIXME: need to be removed when properly handling torch 1.7.1
# short term fix is to override the default pip installed torch
- run:
name: Install Dependencies
command: |
sudo apt-get install -y libopenmpi-dev
pip install --progress-bar off -r requirements-test.txt
pip install --progress-bar off torch==1.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
pip install --progress-bar off git+https://github.com/msbaines/torch_pg.git@c85c96f#egg=torch-pg
python -c 'import torch; print("Torch version:", torch.__version__)'
python -m torch.utils.collect_env
install_repo_cpu: &install_repo_cpu install_repo_cpu: &install_repo_cpu
- run: - run:
name: Install Repository name: Install Repository
...@@ -156,7 +170,7 @@ jobs: ...@@ -156,7 +170,7 @@ jobs:
keys: keys:
- cache-key-cpu-{{ checksum "setup.py"}}-{{ checksum "requirements-test.txt"}} - cache-key-cpu-{{ checksum "setup.py"}}-{{ checksum "requirements-test.txt"}}
- <<: *install_dep_17 - <<: *install_dep_17_cpu
- save_cache: - save_cache:
paths: paths:
...@@ -276,7 +290,7 @@ jobs: ...@@ -276,7 +290,7 @@ jobs:
keys: keys:
- cache-key-gpu17-{{ checksum "setup.py"}}-{{ checksum "requirements-test.txt"}} - cache-key-gpu17-{{ checksum "setup.py"}}-{{ checksum "requirements-test.txt"}}
- <<: *install_dep_17 - <<: *install_dep_17_gpu
- save_cache: - save_cache:
paths: paths:
...@@ -313,7 +327,7 @@ jobs: ...@@ -313,7 +327,7 @@ jobs:
keys: keys:
- cache-key-benchmarks-{{ checksum "setup.py"}}-{{ checksum "requirements-test.txt"}} - cache-key-benchmarks-{{ checksum "setup.py"}}-{{ checksum "requirements-test.txt"}}
- <<: *install_dep_17 - <<: *install_dep_17_gpu
- save_cache: - save_cache:
paths: paths:
......
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