Docker 镜像说明

如何启动

为了让您更便捷地使用 HAT 算法包,我们提供了一键启动 Docker 环境的命令,大幅简化环境搭建流程。 当您获取发布包后,首先进入发布包所在路径,然后根据 Docker容器部署章节的指引,完成基础 Docker 环境安装、将用户添加至 Docker 组以及拉取 Docker 镜像。 完成上述操作后,仅需使用以下命令,就能一键搭建好成功运行 HAT 示例所需的全部环境 。

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

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

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

获取执行脚本

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