From 3450db02e2c5c19690b83adf806a13b4c5c04d78 Mon Sep 17 00:00:00 2001 From: wangxingjun778 Date: Wed, 10 Sep 2025 11:00:06 +0800 Subject: [PATCH] add debug info --- docker/build_image.py | 4 ++++ docker/install.sh | 2 +- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/docker/build_image.py b/docker/build_image.py index 895ec653..cfb583d6 100644 --- a/docker/build_image.py +++ b/docker/build_image.py @@ -299,6 +299,7 @@ class LLMImageBuilder(Builder): f'{self.args.torch_version} {self.args.torchvision_version} {self.args.torchaudio_version} ' f'{self.args.vllm_version} {self.args.lmdeploy_version} {self.args.autogptq_version} ' f'{self.args.flashattn_version}') + print(f'>>version_args: {version_args}') with open('docker/Dockerfile.ubuntu', 'r') as f: content = f.read() content = content.replace('{base_image}', self.args.base_image) @@ -319,7 +320,10 @@ class LLMImageBuilder(Builder): self.args.modelscope_branch) content = content.replace('{swift_branch}', self.args.swift_branch) + print('=' * 100) print(f'>>Debug content for llm-image: \n{content}\n\n') + print('=' * 100) + print('\n\n') return content def image(self) -> str: diff --git a/docker/install.sh b/docker/install.sh index d7fc2b5b..595ddf5f 100644 --- a/docker/install.sh +++ b/docker/install.sh @@ -30,7 +30,7 @@ if [[ "$(printf '%s\n' "0.6.0" "$vllm_version" | sort -V | head -n1)" = "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." + 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