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 dockerfile

Browse Source
This commit is contained in:
yuze.zyz
2024-09-20 15:20:46 +08:00
parent 6d1fdd8747
commit 4973ff6cc3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docker/Dockerfile.ubuntu
View File

@@ -58,8 +58,8 @@ RUN if [ "$USE_GPU" = "True" ] ; then \
pip install --no-cache-dir -U 'xformers<0.0.27' --index-url https://download.pytorch.org/whl/cu121 && \
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 && \
pip install --no-cache-dir 'vllm==0.6.0' 'lmdeploy==0.6.0'; \
pip install --no-cache-dir -U triton 'lmdeploy==0.6.0' && \
pip install --no-cache-dir 'vllm==0.6.0'; \
else \
echo 'cpu unsupport vllm auto-gptq'; \
fi