[chore] move a test script and a CI test improvement (#464)

* [chore] move a test script

* add a shortcut for installing

* more skipping

* keep apt-get part

.circleci/config.yml

@@ -64,6 +64,9 @@ install_dep_151: &install_dep_151
       name: Install Dependencies with torch 1.5.1
       command: |
         sudo apt-get install -y libopenmpi-dev
+        # check if we have restored venv cache (/home/circleci/venv) correctly, if so, just skip
+        python -c 'import torch; assert torch.__version__.split(".")[:2] != ["1", "5"]' || exit 0
+        # start installing
         pip install --progress-bar off torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html
         pip install --progress-bar off -r requirements-test.txt
         pip install --progress-bar off -r requirements-benchmarks.txt
@@ -76,6 +79,9 @@ install_dep_160: &install_dep_160
       name: Install Dependencies with torch 1.6.0
       command: |
         sudo apt-get install -y libopenmpi-dev
+        # check if we have restored venv cache (/home/circleci/venv) correctly, if so, just skip
+        python -c 'import torch; assert torch.__version__.split(".")[:2] != ["1", "6"]' || exit 0
+        # start installing
         pip install --progress-bar off torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
         pip install --progress-bar off -r requirements-test.txt
         pip install --progress-bar off -r requirements-benchmarks.txt
@@ -89,6 +95,9 @@ install_dep_171: &install_dep_171
       name: Install Dependencies with torch 1.7.1
       command: |
         sudo apt-get install -y libopenmpi-dev
+        # check if we have restored venv cache (/home/circleci/venv) correctly, if so, just skip
+        python -c 'import torch; assert torch.__version__.split(".")[:2] != ["1", "7"]' || exit 0
+        # start installing
         pip install --progress-bar off torch==1.7.1+cu101 torchvision==0.8.2+cu101 -f https://download.pytorch.org/whl/torch_stable.html
         pip install --progress-bar off -r requirements-test.txt
         pip install --progress-bar off -r requirements-benchmarks.txt
@@ -104,6 +113,9 @@ install_dep_171_cu110: &install_dep_171_cu110
         sudo add-apt-repository universe
         sudo apt-get update
         sudo apt-get install -y libopenmpi-dev
+        # check if we have restored venv cache (/home/circleci/venv) correctly, if so, just skip
+        python -c 'import torch; assert torch.__version__.split(".")[:2] != ["1", "7"]' || exit 0
+        # start installing
         pip install --progress-bar off torch==1.7.1+cu110 torchvision==0.8.2+cu110 -f https://download.pytorch.org/whl/torch_stable.html
         pip install --progress-bar off -r requirements-test.txt
         pip install --progress-bar off -r requirements-benchmarks.txt
@@ -117,6 +129,9 @@ install_dep_180: &install_dep_180
       name: Install Dependencies with torch 1.8.0 nightly
       command: |
         sudo apt-get install -y libopenmpi-dev
+        # check if we have restored cache correctly, if so, just skip
+        python -c 'import torch; assert torch.__version__.split(".")[:2] != ["1", "8"]' || exit 0
+        # start installing
         pip install --progress-bar off -r requirements-test.txt
         # Since we are using nightly builds, we bypass the benchmarks req file
         # and install ourselves for testing.