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

246 阅读2分钟

欢迎关注我的公众号 [极智视界],获取我的更多笔记分享

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

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

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


 【公众号传送】

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


logo_show.gif