极智开发 | 多用户共用一张昇腾卡开发教程

Posted 2022-11-30 极智视界

  欢迎关注我，获取我的更多笔记分享

  大家好，我是极智视界，本文介绍一下 多用户共用一张昇腾卡开发教程。

  昇腾计算应该是国产化里的首选了，但是用昇腾卡计算的坑也是很多的。比如这里：多用户不能共用一张卡、只能用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号卡计算资源进行计算。

  好了，以上分享了 多用户共用一张昇腾卡开发方法。希望我的分享能对你的学习有一点帮助。