大家好,我是极智视界,本文介绍一下 多用户共用一张昇腾卡开发教程。
昇腾计算应该是国产化里的首选了,但是用昇腾卡计算的坑也是很多的。比如这里:多用户不能共用一张卡、只能用HwHiAiUser进行资源调用 … 这篇让我们把这些问题给咔咔解决掉。
首先基本前提环境是:计算卡 atlas300IPro、操作系统centos7.6、docker开发、CANN5.1.RC2、一个可用的镜像 arm64/centos7.6-cann。
下面开始。
- 构建容器a => 用户名为
HwHiAiUser
=> 使用0号卡
# 生成容器a
docker run --name=container_a \
-p 容器端口号:22 \
--privileged -it \
--device=/dev/davinci0 \
--device=/dev/davinci_manager \
--device=/dev/devmm_svm \
--device=/dev/hisi_hdc \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
-v /宿主机目录:/容器目录 \
arm64/centos7.6-cann /bin/bash
# 配置环境
# 启动容器a
docker start container_a
docker attach container_a
# 这个时候的用户名为 HwHiAiUser,可以直接进行开发操作
- 构建容器b => 用户名为
root
=> 同样使用0号卡
# 生成容器b
docker run --name=container_b \
-p 容器端口号:22 \
-u root \
--privileged -it \
--device=/dev/davinci0 \
--device=/dev/davinci_manager \
--device=/dev/devmm_svm \
--device=/dev/hisi_hdc \
-v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
-v /宿主机目录:/容器目录 \
arm64/centos7.6-cann /bin/bash
# 配置环境
# 启动容器b
docker start container_b
docker attach container_b
# 这个时候的用户名为root,需要配置环境才可进行开发操作
# 安装te
pip install /usr/local/Ascend/ascend-toolkit/latest/aarch64-linux/lib64/te-0.4.0-py3-none-any.whl
# 安装topi
pip install /usr/local/Ascend/ascend-toolkit/latest/aarch64-linux/lib64/topi-0.4.0-py3-none-any.whl
# 配置sdk环境
source /usr/local/Ascend/ascend-toolkit/set_env.sh
# 配置atc环境 -> 构建软链
ln -s /usr/local/Ascend/ascend-toolkit/5.1.RC2/aarch64-linux/bin/atc /usr/bin/
# 这样就可以在root用户下进行模型转换+模型推理了
以上,两个容器docker_a和docker_b可以共同调用0号卡计算资源进行计算。
好了,以上分享了 多用户共用一张昇腾卡开发方法。希望我的分享能对你的学习有一点帮助。
标签:多用户,--,Ascend,dev,usr,device,腾卡,极智,local From: https://blog.51cto.com/u_15531854/5825946