From 56c5f14770fbee795c17a41efd75f6b0b6f3387a Mon Sep 17 00:00:00 2001 From: tastelikefeet Date: Mon, 9 Mar 2026 14:19:37 +0800 Subject: [PATCH] fix --- docker/Dockerfile.ubuntu | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/docker/Dockerfile.ubuntu b/docker/Dockerfile.ubuntu index 84a00a66..47b6633a 100644 --- a/docker/Dockerfile.ubuntu +++ b/docker/Dockerfile.ubuntu @@ -65,13 +65,13 @@ RUN bash /tmp/install.sh {version_args} && \ RUN if [ "$IMAGE_TYPE" = "gpu" ]; then \ - pip install math_verify "gradio<5.33" "deepspeed<0.18" ray -U && \ - pip install liger_kernel wandb swanlab nvitop pre-commit "transformers" "trl<0.25" "peft<0.18" huggingface-hub -U && \ - pip install --no-build-isolation transformer_engine[pytorch]; \ + pip install --no-cache-dir math_verify "gradio<5.33" "deepspeed<0.18" ray -U && \ + pip install --no-cache-dir liger_kernel wandb swanlab nvitop pre-commit "transformers" "trl<0.25" "peft<0.18" huggingface-hub -U && \ + pip install --no-cache-dir --no-build-isolation transformer_engine[pytorch]; \ cd /tmp && GIT_LFS_SKIP_SMUDGE=1 git clone https://github.com/NVIDIA/apex && \ cd apex && pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./ && \ cd / && rm -fr /tmp/apex && pip cache purge; \ - pip install megatron-core==0.15.*; \ + pip install --no-cache-dir "megatron-core==0.15.*"; \ pip uninstall autoawq -y; \ elif [ "$IMAGE_TYPE" = "cpu" ]; then \ pip install --no-cache-dir huggingface-hub transformers peft diffusers -U; \