Added a script for building wheels on a new docker image. (facebookre…

…search#257)

* Added a script for building theseus on a new docker image.

* Add missing newline.

* Fixed dockerfile generation to work with other CUDA versions.

* Changed build script to pass CUDA version as an arg, and added option for CPU-only.

* Changed build script to get tar.gz sources and added also default CUDA version naming.

* Replaced theseus-ai by theseus-opt.

build_scripts/build_wheel.sh

@@ -0,0 +1,80 @@
+#bin/bash
+
+# Ensure that 2 arguments (ROOT_DIR, TAG, CUDA_VERSION) are provided
+die () {
+    echo >&2 "$@"
+    exit 1
+}
+[ "$#" -eq 3 ] || die "3 arguments required, $# provided"
+ROOT_DIR=$1
+TAG=$2
+CUDA_VERSION=$3
+
+CUDA_VERSION_SUPPORTED=$(echo "cpu 10.2 11.3 11.6" | grep -w ${CUDA_VERSION})
+[ "${CUDA_VERSION_SUPPORTED}" ] || die "CUDA_VERSION must be one of (cpu, 10.2, 11.3, 11.6)"
+
+
+CUDA_SUFFIX=$(echo ${CUDA_VERSION} | sed 's/[.]//g')
+
+if [[ ${CUDA_VERSION} == "cpu" ]] 
+then
+    DEVICE_TAG=cpu
+    IMAGE_NAME="pytorch/manylinux-cuda102"
+    GPU_ARGS=""
+else
+    DEVICE_TAG="cu${CUDA_SUFFIX}"
+    IMAGE_NAME="pytorch/manylinux-cuda${CUDA_SUFFIX}"
+    GPU_ARGS=" --gpus all"
+fi
+
+for PYTHON_VERSION in 3.9; do
+    # Create dockerfile to build in manylinux container
+    DOCKER_DIR=${ROOT_DIR}/docker_${PYTHON_VERSION}
+    mkdir -p ${DOCKER_DIR}
+    echo """FROM ${IMAGE_NAME}
+    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
+    RUN which python
+    ENV CUDA_HOME /usr/local/cuda-${CUDA_VERSION}
+    RUN pip install torch --extra-index-url https://download.pytorch.org/whl/${DEVICE_TAG}
+    RUN conda install -c conda-forge suitesparse
+    RUN pip install build wheel
+    RUN git clone https://github.com/facebookresearch/theseus.git
+    WORKDIR theseus
+    RUN git checkout ${TAG} -b tmp_build
+    CMD python3 -m build --no-isolation
+    """ > ${DOCKER_DIR}/Dockerfile
+
+    # Run the container
+    cd ${DOCKER_DIR}
+    echo $(pwd)
+    DOCKER_NAME=theseus_${PYTHON_VERSION}
+    sudo docker build -t "${DOCKER_NAME}_img" .
+    sudo docker run${GPU_ARGS} --name ${DOCKER_NAME} ${DOCKER_NAME}_img
+
+    # Copy the wheel to host
+    CP_STR="cp"$(echo ${PYTHON_VERSION} | sed 's/[.]//g')
+    if [[ ${CUDA_VERSION} == "cpu" ]] 
+    then
+        DOCKER_WHL="theseus/dist/theseus_opt-${TAG}-py3-none-any.whl"
+        HOST_WHL="theseus_opt-${TAG}-py3-none-any.whl"
+    else
+        DOCKER_WHL="theseus/dist/theseus_opt-${TAG}-${CP_STR}-${CP_STR}-linux_x86_64.whl"
+        if [[ ${CUDA_VERSION} == "10.2" ]]
+        then
+            PLUS_CU_TAG=""  # 10.2 will be the pypi version, so don't add +cu102
+        else
+            PLUS_CU_TAG="+${DEVICE_TAG}"
+        fi
+        HOST_WHL="theseus_ai-${TAG}${PLUS_CU_TAG}-${CP_STR}-${CP_STR}-manylinux_2_17_x86_64.whl"
+    fi
+
+    sudo docker cp "${DOCKER_NAME}:theseus/dist/theseus-opt-${TAG}.tar.gz" "${DOCKER_DIR}/theseus-opt-${TAG}.tar.gz"
+    sudo docker cp "${DOCKER_NAME}:${DOCKER_WHL}" ${DOCKER_DIR}/${HOST_WHL}
+    sudo docker rm ${DOCKER_NAME}
+    sudo docker image rm "${DOCKER_NAME}_img"
+done

docs/source/getting-started.rst

@@ -17,7 +17,7 @@ Installing
 ^^^^^^^^^^
 .. code-block:: bash
 
-    pip install theseus-ai
+    pip install theseus-opt
 
 If you are interested in contributing to ``theseus``, instead install
 

setup.py

@@ -58,7 +58,7 @@ def parse_requirements_file(path):
     ext_modules = []
 
 setuptools.setup(
-    name="theseus-ai",
+    name="theseus-opt",
     version=version,
     author="Meta Research",
     description="A library for differentiable nonlinear optimization.",