10.4.1. Docker 镜像说明¶
10.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_x5_gpu:{version}
注解
此处的版本号{version}仅为示例,请将其替换为您实际获取到的镜像版本号。
其中 openexplorer/ai_toolchain_ubuntu_20_x5_gpu:{version}
是镜像名, -v
用于将本地的路径挂载到docker路径下。
10.4.1.2. 获取执行脚本¶
成功启动docker环境之后,从发布包中获取 release_packages
的压缩包,解压之后会看到 configs
和 tools
两个文件夹目录。
所有环境都拥有之后,你就可以按照后面的训练教程来一步步的使用HAT训练出一个定点模型。