Update Dockerfile-web

Author: IAMJOYBO

Dockerfile-web

@@ -9,26 +9,53 @@ ENV DEBIAN_FRONTEND=noninteractive
 WORKDIR /workspace
 ENV CUDA_HOME=/usr/local/cuda
 
-# 配置WEB所需环境
-RUN curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /usr/share/keyrings/nodesource.gpg
-RUN chmod 644 /usr/share/keyrings/nodesource.gpg
-RUN echo "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/nodesource.gpg] https://deb.nodesource.com/node_23.x nodistro main" | tee /etc/apt/sources.list.d/nodesource.list
-
 # 安装依赖
-RUN apt update -y && apt install -y apt-utils && apt upgrade && apt install -y --no-install-recommends net-tools tree vim curl git git-lfs neofetch libtbb-dev libssl-dev libcurl4-openssl-dev libaio1 libaio-dev libfmt-dev libgflags-dev zlib1g-dev patchelf git wget vim gcc g++ apt-transport-https ca-certificates curl gnupg nodejs && apt-get clean && rm -rf /var/lib/apt/lists/*
+RUN apt update -y && apt install -y apt-utils && apt upgrade -y
+RUN apt install -y --no-install-recommends \
+    libtbb-dev \
+    libssl-dev \
+    libcurl4-openssl-dev \
+    libaio1 \
+    libaio-dev \
+    libfmt-dev \
+    libgflags-dev \
+    zlib1g-dev \
+    patchelf \
+    git \
+    wget \
+    vim \
+    gcc \
+    g++
 
 # 安装CMake
 RUN wget https://github.com/Kitware/CMake/releases/download/v4.0.1/cmake-4.0.1-linux-x86_64.sh && echo y | bash cmake-4.0.1-linux-x86_64.sh && rm -rf cmake-4.0.1-linux-x86_64.sh
 
 # 克隆代码
 RUN git clone https://github.com/kvcache-ai/ktransformers.git 
+# 清理 apt 缓存
+RUN rm -rf /var/lib/apt/lists/*
+
 # 进入项目目录
 WORKDIR /workspace/ktransformers
 # 初始化子模块
 RUN git submodule update --init --recursive
 
 # 升级 pip
-RUN pip install --upgrade pip && pip install -U wheel setuptools ninja pyproject numpy cpufeature aiohttp zmq openai flash-attn && pip cache purge
+RUN pip install --upgrade pip && pip install -U wheel setuptools
+
+# 安装构建依赖
+RUN pip install ninja pyproject numpy cpufeature aiohttp zmq openai
+
+# 安装 flash-attn（提前装可以避免后续某些编译依赖出错）
+RUN pip install flash-attn
+
+# 安装WEB所需环境
+RUN apt-get update -y && apt-get install -y apt-transport-https ca-certificates curl gnupg
+RUN curl -fsSL https://deb.nodesource.com/gpgkey/nodesource-repo.gpg.key | gpg --dearmor -o /usr/share/keyrings/nodesource.gpg
+RUN chmod 644 /usr/share/keyrings/nodesource.gpg
+RUN echo "deb [arch=$(dpkg --print-architecture) signed-by=/usr/share/keyrings/nodesource.gpg] https://deb.nodesource.com/node_23.x nodistro main" | tee /etc/apt/sources.list.d/nodesource.list
+RUN apt-get update -y
+RUN apt-get install nodejs -y
 
 # 安装 WEB
 RUN cd /workspace/ktransformers/ktransformers/website/public && rm -rf config.js && wget https://github.com/IAMJOYBO/ktransformers/raw/refs/heads/main/config.js
@@ -41,20 +68,24 @@ RUN CPU_INSTRUCT=${CPU_INSTRUCT} \
     TORCH_CUDA_ARCH_LIST="8.0;8.6;8.7;8.9;9.0+PTX" \
     pip install . --no-build-isolation --verbose
 
-RUN pip install third_party/custom_flashinfer/ && pip cache purge
+RUN pip install third_party/custom_flashinfer/
+# 清理 pip 缓存
+RUN pip cache purge
 
 # 拷贝 C++ 运行时库
 RUN cp /usr/lib/x86_64-linux-gnu/libstdc++.so.6 /opt/conda/lib/
 
+# 安装部分个人使用的工具
+RUN apt install -y net-tools tree vim curl git git-lfs neofetch 
 # apt修改为国内源
 RUN rm -rf /etc/apt/sources.list && wget https://github.com/IAMJOYBO/ktransformers/raw/refs/heads/main/sources-20.04.list -O /etc/apt/sources.list
 RUN pip config set global.index-url https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple
 
 # 预下载的配置文件
-# RUN pip install huggingface_hub modelscope
-# RUN huggingface-cli download deepseek-ai/DeepSeek-R1 --exclude *.safetensors --local-dir /app/model/DeepSeek-R1
-# RUN huggingface-cli download deepseek-ai/DeepSeek-V3-0324 --exclude *.safetensors --local-dir /app/model/DeepSeek-V3-0324
-# RUN huggingface-cli download deepseek-ai/DeepSeek-V2-Lite-Chat --exclude *.safetensors --local-dir /app/model/DeepSeek-V2-Lite-Chat
+RUN pip install huggingface_hub modelscope
+RUN huggingface-cli download deepseek-ai/DeepSeek-R1 --exclude *.safetensors --local-dir /app/model/DeepSeek-R1
+RUN huggingface-cli download deepseek-ai/DeepSeek-V3-0324 --exclude *.safetensors --local-dir /app/model/DeepSeek-V3-0324
+RUN huggingface-cli download deepseek-ai/DeepSeek-V2-Lite-Chat --exclude *.safetensors --local-dir /app/model/DeepSeek-V2-Lite-Chat
 
 # 保持容器运行（调试用）
 ENTRYPOINT ["tail", "-f", "/dev/null"]