更新时间:2024-10-30 GMT+08:00
分享

标注镜像Dockerfile示例

一般情况下,引擎主要包含预标注算法或预审核算法运行所需要的基本依赖环境,用户也可将预标注算法或预审核算法包内置在AI引擎中。

用户可使用命令行模式或Dockerfile模式进行构建。

以预标注自定义镜像为例,一般的镜像制作Dockerfile示例如下(xxx替换为实际路径):

# 载入基础镜像,训练或评测引擎一般需包含cuda/cudnn等算法基础环境。用户可手动制作或拉取官方镜像
FROM xxx/cuda:11.0.3-devel-ubuntu18.04
 
# 设置工作目录【可选】默认为ROOT,用户可修改USER及PATH
WORKDIR /root/workspace
 
# 如果是内置预标注算法的自定义镜像,需要把预标注算法复制到工作目录下,
COPY /path/to/algorithom /path/to/algorithom
 
# 按需安装用户APT环境。如果需要修改/etc/apt/sources.list可替换
COPY /path/to/sources.list /etc/apt/sources.list
RUN apt-get install vim
 
# 按需安装用户算法环境。如果需要修改~/.pip/pip.conf可替换。用户也可安装miniconda进行包管理
COPY /path/to/pip.conf /root/.pip/pip.conf
COPY /path/to/requirements.txt /root
RUN pip install –r /root/requirements.txt
 
# 设置环境变量【可选】
ENV PYTHONUNBUFFERED 1

对于Dockerfile的统一构建方式如下:

docker build –f [DockerfileName] –t [ImageName:ImageVersion] .

相关文档