From bd68182d342e53a2a128de2f93fdd2fb05b509b1 Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Thu, 7 Mar 2024 14:32:05 +0800 Subject: [PATCH] fix pip install tensorboardX --- .dev_scripts/build_image.sh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.dev_scripts/build_image.sh b/.dev_scripts/build_image.sh index 5c3e0915..114eb7db 100644 --- a/.dev_scripts/build_image.sh +++ b/.dev_scripts/build_image.sh @@ -155,7 +155,7 @@ docker_file_content=`cat docker/Dockerfile.ubuntu` BUILD_HASH_ID=$(git rev-parse HEAD) # install thrid part library -docker_file_content="${docker_file_content} \nRUN export COMMIT_ID=$BUILD_HASH_ID && pip install --no-cache-dir -U adaseq pai-easycv https://modelscope-open.oss-cn-hangzhou.aliyuncs.com/ms_swift-1.7.0-py3-none-any.whl 'funasr==1.0.14' autoawq timm 'transformers==4.38.2' tensorboardX" +docker_file_content="${docker_file_content} \nRUN export COMMIT_ID=$BUILD_HASH_ID && pip install --no-cache-dir -U adaseq pai-easycv https://modelscope-open.oss-cn-hangzhou.aliyuncs.com/ms_swift-1.7.0-py3-none-any.whl 'funasr==1.0.14' autoawq timm 'transformers==4.38.2'" docker_file_content="${docker_file_content} \nRUN pip uninstall modelscope -y && export COMMIT_ID=$BUILD_HASH_ID && cd /tmp && GIT_LFS_SKIP_SMUDGE=1 git clone -b $build_branch --single-branch $REPO_URL && cd modelscope && pip install . && cd / && rm -fr /tmp/modelscope && pip cache purge;" @@ -166,7 +166,7 @@ else echo "Building dsw image will need set ModelScope lib cache location." docker_file_content="${docker_file_content} \nENV MODELSCOPE_CACHE=/mnt/workspace/.cache/modelscope" # pre compile extension - docker_file_content="${docker_file_content} \nRUN pip uninstall -y tb-nightly tensorboardX tensorboard && 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()'" + docker_file_content="${docker_file_content} \nRUN pip uninstall -y tb-nightly tensorboard && 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 docker_file_content="${docker_file_content} \n RUN cp /tmp/resources/conda.aliyun ~/.condarc && \