Update Dockerfile-web

Author: IAMJOYBO

Dockerfile-web

@@ -9,53 +9,26 @@ 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 -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++
+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/*
 
 # 安装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
-
-# 安装构建依赖
-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
+RUN pip install --upgrade pip && pip install -U wheel setuptools ninja pyproject numpy cpufeature aiohttp zmq openai flash-attn && pip cache purge
 
 # 安装 WEB
 RUN cd /workspace/ktransformers/ktransformers/website/public && rm -rf config.js && wget https://github.com/IAMJOYBO/ktransformers/raw/refs/heads/main/config.js
@@ -68,24 +41,20 @@ 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 缓存
-RUN pip cache purge
+RUN pip install third_party/custom_flashinfer/ && 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"]