Skip to content

Commit

Permalink
Removed CUDA 10.2 wheel job from CI.
Browse files Browse the repository at this point in the history
  • Loading branch information
luisenp committed Nov 8, 2022
1 parent 256c483 commit db77552
Show file tree
Hide file tree
Showing 2 changed files with 18 additions and 31 deletions.
18 changes: 3 additions & 15 deletions .circleci/config.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 python3.9 python3.9-dev
sudo apt-get install -y python3.9 python3.9-dev
install_nox: &install_nox
- run:
Expand Down Expand Up @@ -290,26 +290,15 @@ jobs:
- <<: *setup_project
- <<: *run_tests

test_cuda10_wheel:
executor: gpu_cuda10
steps:
- checkout
- <<: *update_and_install_python
- <<: *install_suitesparse
- <<: *setup_cuda11_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
- <<: *build_cuda11_wheel
- <<: *install_suitesparse
- <<: *setup_environment
- <<: *run_tests_from_wheel


Expand All @@ -323,5 +312,4 @@ workflows:
- py39_linux
- unittests_gpu17_cuda10
- unittests_gpu17_cuda11
- test_cuda10_wheel
- test_cuda11_wheel
31 changes: 15 additions & 16 deletions build_scripts/build_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -76,21 +76,6 @@ for PYTHON_VERSION in 3.9; do
echo """# ----------------
FROM ${IMAGE_NAME}
# --- Install conda and environment
ENV CONDA_DIR /opt/conda
RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
/bin/bash ~/miniconda.sh -b -p /opt/conda
ENV PATH \$CONDA_DIR/bin:\$PATH
RUN conda create --name theseus python=${PYTHON_VERSION}
RUN source activate theseus
# --- Install torch
ENV CUDA_HOME /usr/local/cuda-${CUDA_VERSION}
RUN pip install ${TORCH_VERSION} --extra-index-url https://download.pytorch.org/whl/${DEVICE_TAG}
# --- Install sparse suitesparse
RUN conda install -c conda-forge suitesparse
# --- Install baspacho dependencies (cmake, BLAS)
RUN wget --quiet https://github.com/Kitware/CMake/releases/download/v3.24.2/cmake-3.24.2-linux-x86_64.sh -O ~/cmake3.24.sh
RUN mkdir /opt/cmake3.24
Expand All @@ -101,7 +86,6 @@ for PYTHON_VERSION in 3.9; do
# --- Install baspacho
RUN git clone https://github.com/facebookresearch/baspacho.git
WORKDIR baspacho
# Note: to use static BLAS the option is really BLA_STATIC (https://cmake.org/cmake/help/latest/module/FindBLAS.html)
RUN /opt/cmake3.24/bin/cmake -S . -B build -DCMAKE_BUILD_TYPE=Release -DBLA_STATIC=ON \
${BASPACHO_CUDA_ARGS} \
Expand All @@ -110,6 +94,21 @@ for PYTHON_VERSION in 3.9; do
RUN /opt/cmake3.24/bin/cmake --build build -- -j16
WORKDIR ..
# --- Install conda and environment
ENV CONDA_DIR /opt/conda
RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
/bin/bash ~/miniconda.sh -b -p /opt/conda
ENV PATH \$CONDA_DIR/bin:\$PATH
RUN conda create --name theseus python=${PYTHON_VERSION}
RUN source activate theseus
# --- Install torch
ENV CUDA_HOME /usr/local/cuda-${CUDA_VERSION}
RUN pip install ${TORCH_VERSION} --extra-index-url https://download.pytorch.org/whl/${DEVICE_TAG}
# --- Install sparse suitesparse
RUN conda install -c conda-forge suitesparse
# --- Compile theseus wheel
RUN pip install build wheel
RUN git clone https://github.com/facebookresearch/theseus.git
Expand Down

0 comments on commit db77552

Please sign in to comment.