46 lines
1.3 KiB
Docker
46 lines
1.3 KiB
Docker
|
|
FROM registry.d-robotics.cc/public/cuda:11.8.0-cudnn8-devel-ubuntu22.04
|
|
# ccr-29eug8s3-pub.cnc.bj.baidubce.com/public/cuda:11.8.0-cudnn8-devel-ubuntu22.04
|
|
WORKDIR /app
|
|
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
ENV PYTHONUNBUFFERED=1
|
|
ENV TZ=Asia/Shanghai
|
|
|
|
RUN sed -i 's/archive.ubuntu.com/mirrors.tuna.tsinghua.edu.cn/g' /etc/apt/sources.list && \
|
|
sed -i 's/security.ubuntu.com/mirrors.tuna.tsinghua.edu.cn/g' /etc/apt/sources.list
|
|
|
|
RUN apt-get update && apt-get install -y \
|
|
software-properties-common \
|
|
&& add-apt-repository ppa:deadsnakes/ppa \
|
|
&& apt-get update \
|
|
&& apt-get install -y \
|
|
python3.10 \
|
|
python3.10-dev \
|
|
python3.10-distutils \
|
|
libgl1-mesa-glx \
|
|
libglib2.0-0 \
|
|
wget \
|
|
ffmpeg \
|
|
libsm6 \
|
|
libxext6 \
|
|
&& rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1
|
|
|
|
COPY . /app/
|
|
|
|
RUN python3 -m pip install --upgrade pip
|
|
|
|
RUN pip install torch==2.1.0 torchvision==0.16.0 --index-url https://download.pytorch.org/whl/cu121
|
|
|
|
RUN pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
|
|
|
|
RUN pip install packaging==24.0
|
|
|
|
RUN pip install flash_attn-2.7.2.post1+cu12torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl
|
|
|
|
RUN mkdir -p /app/dataset/input /app/dataset/output
|
|
|
|
ENTRYPOINT ["bash", "finetune.sh"]
|