modelscope/docker/scripts/install_pytorch3d_nvdiffrast.sh

export CMAKE_BUILD_PARALLEL_LEVEL=36 \
        && export MAX_JOBS=36 \
        && export CMAKE_CUDA_ARCHITECTURES="50;52;60;61;70;75;80;8.6+PTX;87;89;90" \
        && export TORCH_CUDA_ARCH_LIST="5.0;5.2;6.0;6.1;7.0;7.5;8.0;8.6+PTX;8.7;8.9;9.0" \
        && git clone --branch 2.1.0 --recursive https://github.com/NVIDIA/thrust.git \
        && cd thrust \
        && mkdir build \
        && cd build \
        && cmake -DCMAKE_INSTALL_PREFIX=/usr/local/cuda/ -DTHRUST_INCLUDE_CUB_CMAKE=ON .. \
        && make install \
        && cd ../.. \
        && rm -rf thrust \
        && pip install --no-cache-dir fvcore iopath \
	&& curl -LO https://github.com/NVIDIA/cub/archive/2.1.0.tar.gz \
        && tar xzf 2.1.0.tar.gz \
        && export CUB_HOME=$PWD/cub-2.1.0 \
        && FORCE_CUDA=1 pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable" \
        && rm -fr 2.1.0.tar.gz $PWD/cub-2.1.0 \
        && apt-get update \
        && apt-get install -y --no-install-recommends pkg-config libglvnd0 libgl1 libglx0 libegl1  libgles2 libglvnd-dev libgl1-mesa-dev libegl1-mesa-dev  libgles2-mesa-dev -y \
        && git clone https://github.com/NVlabs/nvdiffrast.git \
        && cd nvdiffrast \
        && pip install --no-cache-dir . \
        && cd .. \
        && rm -rf nvdiffrast