Dockerfile

# 使用官方的、预先配置好的PaddlePaddle GPU镜像作为基础
# 这保证了CUDA, cuDNN, Python, paddlepaddle-gpu, numpy之间100%兼容
FROM paddlepaddle/paddle:2.6.1-gpu-cuda11.8-cudnn8-runtime

# 设置工作目录
WORKDIR /app

# 安装一些系统库，以防止其他Python库安装失败
RUN apt-get update && apt-get install -y libgl1-mesa-glx libglib2.0-0 && rm -rf /var/lib/apt/lists/*

# 复制简化的依赖文件
COPY requirements.txt .

# 安装剩余的、安全的依赖库
# paddlepaddle-gpu和numpy已经完美存在于基础镜像中，不会被破坏
RUN pip install --no-cache-dir -r requirements.txt

# 复制应用代码
COPY app.py .

# 设置启动命令
CMD ["gradio", "app.py", "--server-name", "0.0.0.0"]