Update Dockerfile

Author: IAMJOYBO

Dockerfile

@@ -1,22 +1,33 @@
-FROM pytorch/pytorch:2.6.0-cuda12.6-cudnn9-devel
-
-WORKDIR /app/index-tts
+FROM nvcr.io/nvidia/cuda:12.6.3-cudnn-devel-ubuntu24.04
 ENV TZ=Asia/Shanghai
+ENV DEBIAN_FRONTEND=noninteractive
+COPY README.txt ./
+
+RUN apt update && apt install -y build-essential cmake ninja-build patchelf wget net-tools curl iputils-ping git git-lfs
+
+RUN mkdir -p /app
+WORKDIR /app
+RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && bash Miniconda3-latest-Linux-x86_64.sh -b -p /app/conda
 
-RUN apt update && apt install -y wget net-tools tree curl && wget https://github.com/index-tts/index-tts/raw/refs/heads/main/requirements.txt
-RUN  pip install -r requirements.txt && pip install deepspeed
+RUN /app/conda/bin/conda init
+RUN conda create --name ktransformers python=3.11
+SHELL ["conda", "run", "-n", "ktransformers", "/bin/bash", "-c"]
+RUN conda activate ktransformers
+RUN conda install -c conda-forge libstdcxx-ng && strings ~/anaconda3/envs/ktransformers/lib/libstdc++.so.6 | grep GLIBCXX
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
+RUN pip install packaging ninja cpufeature numpy
 
-RUN apt update && apt install -y ffmpeg gcc g++ cmake
+RUN apt install -y libtbb-dev libssl-dev libcurl4-openssl-dev libaio1 libaio-dev libgflags-dev zlib1g-dev libfmt-dev libnuma-dev
 
-RUN pip install -U triton --index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/Triton-Nightly/pypi/simple/triton-nightly
+RUN git clone https://github.com/kvcache-ai/ktransformers.git && cd ktransformers && git submodule update --init --recursive
+RUN cd ktransformers && bash install.sh
 
-ENV CUDA_HOME=/usr/local/cuda-12.4
-ENV PATH=$CUDA_HOME/bin:$PATH
-ENV LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
+RUN pip install libstdcxx-ng
+RUN pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126
 
-# 设置国内源
-RUN mkdir -p /app/index-tts && rm -rf /etc/apt/sources.list && rm -rf /etc/apt/sources.list.d/*ubuntu*
-COPY sources-22.04.list /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-V2-Lite-Chat --exclude *.safetensors --local-dir /app/model/deepseek-ai/DeepSeek-V2-Lite-Chat
+RUN huggingface-cli download deepseek-ai/DeepSeek-V3-0324 --exclude *.safetensors --local-dir /app/model/deepseek-ai/DeepSeek-V3-0324
+RUN huggingface-cli download deepseek-ai/DeepSeek-R1 --exclude *.safetensors --local-dir /app/model/deepseek-ai/DeepSeek-R1
 
-ENTRYPOINT ["python", "webui.py"]
+CMD tail -f README.txt