mirror of
https://github.com/linyqh/NarratoAI.git
synced 2025-12-10 18:02:51 +00:00
32 lines
890 B
Plaintext
32 lines
890 B
Plaintext
ARG BASE=nvidia/cuda:12.1.0-devel-ubuntu22.04
|
|
FROM ${BASE}
|
|
|
|
# 设置环境变量
|
|
ENV http_proxy=http://host.docker.internal:7890
|
|
ENV https_proxy=http://host.docker.internal:7890
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
# 安装系统依赖
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
gcc g++ make git python3 python3-dev python3-pip python3-venv python3-wheel \
|
|
espeak-ng libsndfile1-dev nano vim unzip wget xz-utils && \
|
|
apt-get clean && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# 设置工作目录
|
|
WORKDIR /root/MiniCPM-V/
|
|
|
|
# 安装 Python 依赖
|
|
RUN git clone https://github.com/OpenBMB/MiniCPM-V.git && \
|
|
cd MiniCPM-V && \
|
|
pip3 install decord && \
|
|
pip3 install --no-cache-dir -r requirements.txt && \
|
|
pip3 install flash_attn
|
|
|
|
# 清理代理环境变量
|
|
ENV http_proxy=""
|
|
ENV https_proxy=""
|
|
|
|
# 设置 PYTHONPATH
|
|
ENV PYTHONPATH="/root/MiniCPM-V/"
|