mirror of
https://github.com/modelscope/modelscope.git
synced 2025-12-16 16:27:45 +01:00
a780935317
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/13408751 * [to #51122016]feat: build image support torch1.x 2.x cuda11.3.0 11.7.1 * remove base version * env cudatookit-version * build script merge * torch 1.11.0 fix
15 lines
814 B
Bash
15 lines
814 B
Bash
export CMAKE_BUILD_PARALLEL_LEVEL=36 && export MAX_JOBS=4 && export CMAKE_CUDA_ARCHITECTURES="50;52;60;61;70;75;80;86" \
|
|
&& pip install --no-cache-dir fvcore iopath \
|
|
&& curl -LO https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz \
|
|
&& tar xzf 1.10.0.tar.gz \
|
|
&& export CUB_HOME=$PWD/cub-1.10.0 \
|
|
&& pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable" \
|
|
&& rm -fr 1.10.0.tar.gz cub-1.10.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
|