Files
modelscope/docker/scripts/install_tiny_cuda_nn.sh
mulin.lyh 86128cd3a8 [to #47612577]feat: image supoort 3d models, python3.8 test version
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11542009


* mmcv-full<=1.7.0 for mmdet3d

* fix code style issue

* add speechbrain==0.5.7

* add compatible comments
2023-02-07 02:28:20 +00:00

9 lines
379 B
Bash

export CMAKE_BUILD_PARALLEL_LEVEL=36 && export MAX_JOBS=36 && export TCNN_CUDA_ARCHITECTURES="50;52;60;61;70;75;80;86" \
&& git clone --recursive https://github.com/nvlabs/tiny-cuda-nn \
&& cd tiny-cuda-nn \
&& git checkout v1.6 \
&& cd bindings/torch \
&& python setup.py install \
&& cd ../../.. \
&& rm -rf tiny-cuda-nn