Fixed errors in CI job for wheels.

.circleci/config.yml

@@ -32,7 +32,7 @@ update_and_install_python: &update_and_install_python
       command: |
         sudo add-apt-repository -y ppa:deadsnakes/ppa
         sudo apt-get update
-        sudo apt-get install -y python3.7 python3.7-dev python3.8 python3.8-dev
+        sudo apt-get install -y python3.7 python3.7-dev python3.8 python3.8-dev python3.9 python3.9-dev
 
 install_nox: &install_nox
   - run:
@@ -86,7 +86,7 @@ setup_environment: &setup_environment
       name: Setup virtualenv and tools
       working_directory: ~/project
       command: |
-        virtualenv ~/theseus_venv -p /usr/bin/python3
+        virtualenv ~/theseus_venv -p /usr/bin/python3.9
         echo ". ~/theseus_venv/bin/activate" >> $BASH_ENV
         . ~/theseus_venv/bin/activate
         pip install --progress-bar off --upgrade pip
@@ -176,6 +176,17 @@ run_tests: &run_tests
         pytest -s tests/test_theseus_layer.py
         pytest -s tests -m "cudaext"
 
+build_cuda10_wheel: &build_cuda10_wheel
+  - run:
+      name: Building wheel for CUDA 10
+      working_directory: /home/circleci
+      command: |
+        THESEUS_GIT_COMMIT=$(git --git-dir project/.git log --format="%H" -n 1)
+        THESEUS_VERSION=$(grep -Eo  "[0-9].[0-9].[0-9][.0-9a-z]*" project/theseus/__init__.py)
+        ./project/build_scripts/build_wheel.sh . ${THESEUS_GIT_COMMIT} 10.2 ${THESEUS_VERSION}
+        pip install $(ls */*.whl)
+        pip install -r project/requirements/dev.txt
+
 build_cuda11_wheel: &build_cuda11_wheel
   - run:
       name: Building wheel for CUDA 11
@@ -184,6 +195,17 @@ build_cuda11_wheel: &build_cuda11_wheel
         THESEUS_GIT_COMMIT=$(git --git-dir project/.git log --format="%H" -n 1)
         THESEUS_VERSION=$(grep -Eo  "[0-9].[0-9].[0-9][.0-9a-z]*" project/theseus/__init__.py)
         ./project/build_scripts/build_wheel.sh . ${THESEUS_GIT_COMMIT} 11.6 ${THESEUS_VERSION}
+        pip install $(ls */*.whl)
+        pip install -r project/requirements/dev.txt
+
+run_tests_from_wheel: &run_tests_from_wheel
+  - run:
+      name: Installing theseus from wheel and running tests
+      working_directory: ~/project
+      command: |
+          mv theseus theseus_tmp
+          python -m pytest tests/test_theseus_layer.py
+          pytest -s tests -m "cudaext"
 
 
 # -------------------------------------------------------------------------------------
@@ -268,15 +290,27 @@ jobs:
       - <<: *setup_project
       - <<: *run_tests
 
+  test_cuda10_wheel:
+    executor: gpu_cuda10
+    steps:
+      - checkout
+      - <<: *update_and_install_python
+      - <<: *install_suitesparse
+      - <<: *setup_cuda10_libs
+      - <<: *setup_environment
+      - <<: *build_cuda10_wheel
+      - <<: *run_tests_from_wheel
+
   test_cuda11_wheel:
     executor: gpu_cuda11
     steps:
       - checkout
       - <<: *update_and_install_python
+      - <<: *install_suitesparse
       - <<: *setup_cuda11_libs
       - <<: *setup_environment
-      - <<: *install_torch_cuda11
       - <<: *build_cuda11_wheel
+      - <<: *run_tests_from_wheel