Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-02-24 20:19:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

add tiny_cuda_nn build

Browse Source
This commit is contained in:
mulin.lyh
2024-01-08 10:10:21 +08:00
parent 90d8197acd
commit 7fdb97e945
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
.dev_scripts/build_image.sh
View File

@@ -177,6 +177,13 @@ else
# pre compile extension
docker_file_content="${docker_file_content} \nRUN pip uninstall -y tb-nightly && pip install --no-cache-dir -U tensorboard && TORCH_CUDA_ARCH_LIST='6.0 6.1 7.0 7.5 8.0 8.9 9.0 8.6+PTX' python -c 'from modelscope.utils.pre_compile import pre_compile_all;pre_compile_all()'"
fi
# install here for easycv extension conflict.
docker_file_content="${docker_file_content} \nRUN if [ \"$USE_GPU\" = \"True\" ] ; then \
bash /tmp/install_tiny_cuda_nn.sh; \
else \
echo 'cpu unsupport tiny_cuda_nn'; \
fi"
if [ "$is_ci_test" == "True" ]; then
echo "Building CI image, uninstall modelscope"
docker_file_content="${docker_file_content} \nRUN pip uninstall modelscope -y"