FROM image.sourcefind.cn:5000/gpu/admin/base/jupyterlab-pytorch:2.2.0-py3.10-cuda12.1-ubuntu22.04-devel
ARG IMAGE=voicechat2
ARG IMAGE_UPPER=voicechat2
ARG BRANCH=gpu
RUN cd /root && git clone -b $BRANCH http://developer.hpccube.com/codes/chenpangpang/$IMAGE.git
WORKDIR /root/$IMAGE/$IMAGE_UPPER
RUN apt-get update && apt-get install -y build-essential  byobu curl wget espeak-ng ffmpeg libopus0 libopus-dev
RUN pip install -r requirements.txt
RUN cd llama.cpp && make GGML_CUDA=1 -j
COPY chenyh/$IMAGE/Meta-Llama-3-8B-Instruct-Q4_K_M.gguf /root/$IMAGE/$IMAGE_UPPER/llama.cpp/
COPY chenyh/$IMAGE/tts_models--en--vctk--vits /root/.local/share/tts/
COPY chenyh/$IMAGE/distil-whisper/distil-large-v2 /root/$IMAGE/$IMAGE_UPPER/