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

fix tinycudann

Browse Source
This commit is contained in:
yuze.zyz
2024-10-22 12:58:22 +08:00
parent eac004d7f2
commit 1478dd7435
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
docker/Dockerfile.ubuntu_base
View File

@@ -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 []

2
docker/install.sh
View File

@@ -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