From 1478dd7435d44505091bd4acdd1962030da90d46 Mon Sep 17 00:00:00 2001 From: "yuze.zyz" Date: Tue, 22 Oct 2024 12:58:22 +0800 Subject: [PATCH] fix tinycudann --- docker/Dockerfile.ubuntu_base | 7 +++++++ docker/install.sh | 2 -- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile.ubuntu_base b/docker/Dockerfile.ubuntu_base index ebad61d9..1cccf754 100644 --- a/docker/Dockerfile.ubuntu_base +++ b/docker/Dockerfile.ubuntu_base @@ -203,4 +203,11 @@ ARG TENSORFLOW_VERSION={tf_version} cd /tmp && git clone -b ms_build --single-branch https://github.com/tastelikefeet/mmcv.git && cd mmcv && MMCV_WITH_OPS=1 MAX_JOBS=32 python setup.py bdist_wheel && cd / && rm -fr /tmp/mmcv && pip cache purge; \ fi + # This limits the cuda121 version + RUN if [ "$USE_GPU" = "True" ] ; then \ + pip install --no-cache-dir --force tinycudann==1.7 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html; \ + else \ + echo 'cpu not install tinycudann'; \ + fi + ENTRYPOINT [] diff --git a/docker/install.sh b/docker/install.sh index 54a33566..a4dddf10 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -35,8 +35,6 @@ cd /tmp && git clone https://github.com/Dao-AILab/flash-attention.git && cd flas pip install --no-cache-dir auto-gptq==$autogptq_version -pip install --no-cache-dir --force tinycudann==1.7 -f https://modelscope.oss-cn-beijing.aliyuncs.com/releases/repo.html - # pip uninstall -y torch-scatter && TORCH_CUDA_ARCH_LIST="6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.9;9.0" pip install --no-cache-dir -U torch-scatter pip install --no-cache-dir -U triton