RecSDK构建镜像(使用Dockerfile)
前提
- 物理机上已经安装好对应CANN版本的驱动和固件
- 物理机上已经安装docker,并且docker网络可用
- 准备基础OS镜像:在物理机上使用命令
- debian: x86架构:
docker pull debian:12从Dockerhub上拉取镜像
- debian: x86架构:
构建步骤
Step1:新建build_images目录。
Step2:在build_images目录下准备CANN包。用户可以从昇腾社区下载9.0.0版本的toolkit包与算子包。用户也可根据实际情况选择其他CANN包。
CANN 8.5.0及之后版本,算子包名称存在变化,下载算子包时注意下载对应名称软件包。
| 版本/包名 | CANN 8.5.0之前 | CANN 8.5.0及之后 |
|---|---|---|
| toolkit包名称 | Ascend-cann-toolkit_{version}_linux-{arch}.run | Ascend-cann-toolkit_{version}_linux-{arch}.run |
| 算子包名称 | Ascend-cann-kernels-{version}_linux-{arch}.run | Ascend-cann-{chip_type}-ops_{version}_linux-{arch}.run |
注意
- 当前Dockerfile中,适配的是CANN 8.5.0及之后版本的算子包名称。
- 如需安装CANN 8.5.0之前版本CANN包,需将Dockerfile中KERNEL_PKG的值修改为:
Ascend-cann-kernels*.run。
安装CANN包需要两个文件,分别是
- version.info(驱动版本文件)
- ascend_install.info(固件驱动安装文件)
可将物理机上相应的文件拷贝到build_images目录。物理机安装驱动与固件后,version.info默认安装路径为/usr/local/Ascend/driver/version.info;
ascend_install.info默认安装路径为/etc/ascend_install.info。
Step3:将Dockerfile移动到build_images目录中,并运行下面命令构建镜像。构建镜像的步骤在Dockerfile中有详细的说明,注释部分是安装CANN包与torchrec相关包的操作。
# 服务器能访问外网
docker build -t recsdk_torch_base:v2.0-[x86|arm] -f Dockerfile_debian .
# 服务器配置代理访问外网
docker build -t recsdk_torch_base:v2.0-[x86|arm] -f Dockerfile_debian --build-arg http_proxy=http://your_proxy --build-arg https_proxy=https://your_proxy .
注意:
- 制作镜像时需确保服务器能访问外网,否则需要配置代理。
- 运行命令前注意修改Dockerfile中的基础镜像名。