mirror of
https://github.com/JefferyHcool/BiliNote.git
synced 2026-05-07 06:02:52 +08:00
17 lines
510 B
Docker
17 lines
510 B
Docker
FROM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04
|
|
|
|
RUN apt update && \
|
|
apt install -y ffmpeg python3-pip && \
|
|
apt clean all && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# 设置 Hugging Face 镜像源环境变量
|
|
ENV HF_ENDPOINT=https://hf-mirror.com
|
|
|
|
WORKDIR /app
|
|
COPY ./backend /app
|
|
RUN pip install --no-cache-dir -i https://pypi.mirrors.ustc.edu.cn/simple -r requirements.txt
|
|
RUN pip install --no-cache-dir -i https://pypi.mirrors.ustc.edu.cn/simple 'transformers[torch]>=4.23'
|
|
|
|
CMD ["python3", "main.py"]
|