File size: 707 Bytes
cab74fb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 | FROM nvidia/cuda:12.8.1-cudnn-devel-ubuntu22.04
WORKDIR /app
COPY ./requirements.txt /app/
RUN apt-get update && apt-get install -y \
python3.10 \
python3.10-dev \
curl \
sox \
openssh-server \
ffmpeg \
libgl1-mesa-glx \
git \
ninja-build \
&& rm -rf /var/lib/apt/lists/*
# 安装 pip
RUN curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py \
&& python3.10 get-pip.py \
&& rm get-pip.py
RUN pip install -r requirements.txt
RUN pip install git+https://github.com/huggingface/transformers@v4.51.3-Qwen2.5-Omni-preview
RUN pip install flash-attn --no-build-isolation
# alias python3 as python
RUN ln -s /usr/bin/python3 /usr/bin/python
CMD ["/bin/bash"] |