"...git@developer.sourcefind.cn:kecinstone/2024-pra-vllm.git" did not exist on "03ffd0a02251e10c1aa14fca8cb0ab1e4e40b886"
Dockerfile 614 Bytes
Newer Older
wangzhengtao's avatar
wangzhengtao committed
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
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 flash-attn --no-build-isolation

# alias python3 as python
RUN ln -s /usr/bin/python3 /usr/bin/python

CMD ["/bin/bash"]