From 7f994b908cfeaf39d9ddb12fc57f629ff3094acf Mon Sep 17 00:00:00 2001 From: wangxingjun778 Date: Mon, 15 Sep 2025 21:50:16 +0800 Subject: [PATCH] update install.sh for llm --- docker/install.sh | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/docker/install.sh b/docker/install.sh index 853e2025..3b88f396 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -27,12 +27,12 @@ pip install --no-cache-dir triton auto-gptq==$autogptq_version -U && pip cache p pip install --no-cache-dir vllm==$vllm_version && pip cache purge -if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "0.6.0" ]]; then - # vllm_version is >= 0.6.0 - echo ">>Debug: Condition met: vllm_version >= 0.6.0, proceeding with installation..." - pip install --no-cache-dir vllm==$vllm_version && pip cache purge -else - echo ">>Debug: Condition not met: vllm_version < 0.6.0, skipping installation. (vllm_version = $vllm_version)" -fi +#if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "0.6.0" ]]; then +# # vllm_version is >= 0.6.0 +# echo ">>Debug: Condition met: vllm_version >= 0.6.0, proceeding with installation..." +# pip install --no-cache-dir vllm==$vllm_version && pip cache purge +#else +# echo ">>Debug: Condition not met: vllm_version < 0.6.0, skipping installation. (vllm_version = $vllm_version)" +#fi # 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