10.1.4.1. Docker 镜像说明

10.1.4.1.1. 如何启动

为方便使用HAT算法包,我们提供了一键启动docker环境的命令,以简化使用过程中的环境搭建步骤。用户在获取发布包后先进入发布包所在的路径, 然后根据 Docker容器部署 一节的内容完成基础Docker环境安装、添加Docker组用户、拉取Docker镜像之后, 只需使用以下命令,就可以获取成功运行HAT示例的所有环境。

nvidia-docker run -it --shm-size="15g" -v `pwd`:/open_explorer openexplorer/ai_toolchain_ubuntu_20_j5_gpu:{version}

注解

此处的版本号{version}仅为示例,请将其替换为您实际获取到的镜像版本号。

其中 openexplorer/ai_toolchain_ubuntu_20_j5_gpu:{version} 是镜像名, -v 用于将本地的路径挂载到docker路径下。

10.1.4.1.2. 获取执行脚本

成功启动docker环境之后,从发布包中获取 release_packages 的压缩包,解压之后会看到 configstools 两个文件夹目录。 所有环境都拥有之后,你就可以按照后面的训练教程来一步步的使用HAT训练出一个定点模型。