Nvidia Docker 真实项目镜像打包

一、环境

• 安装Nvidia显卡驱动
• Docker和Nvidia Docker安装教程

二、导出开发环境pip的依赖

pip3 freeze > requirements.txt

下载pip依赖包（有网主机进行）

pip download -d packages -r requirements.txt

三、Dockerfile

# 设置基础镜像
FROM nvidia/cuda/python3.8:numpy
# 创建者信息
MAINTAINER Mr.zhang

# 设置环境变量，所有操作都是非交互式的
ENV DEBIAN_FRONTEND noninteractive

RUN apt-key del 7fa2af80
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu2004/x86_64/7fa2af80.pub

# 安装 ffmpeg 等依赖
RUN apt update
RUN apt install libgl1-mesa-glx ffmpeg libsm6 libxext6 -y

# 设置时区
RUN echo "Asia/Shanghai" > /etc/timezone
RUN rm -f /etc/localtime
RUN dpkg-reconfigure -f noninteractive tzdata

# 拷贝pip依赖包到镜像中
COPY packages /packages
# 离线安装pip依赖包
RUN pip install --no-index --find-links=/packages -r /packages/requirements.txt
# 安装完成删除pip依赖包
RUN rm -rf /packages

# 拷贝项目代码到进行中
COPY lettin_cvess /lettin_cvess