昇腾Atlas Docker开发环境搭建教程

释放双眼,带上耳机,听听看~!
本教程详细介绍了在宿主机上安装昇腾的驱动,并从基础镜像开始构建昇腾Atlas Docker开发环境,包括创建容器、安装cann-toolkit、创建HwHiAiUser用户组、安装Python3.7.5环境等步骤。适合对昇腾开发环境搭建感兴趣的开发者阅读。

本文正在参加t.zsxq.com/0aiNxERDq

先在宿主机安装昇腾的驱动,可以参考我的这篇《昇腾开发环境搭建》。

在宿主机上确保 npu-smi info 能正常输出卡的信息。

下面开始从基础镜像开始构建昇腾atlas docker开发环境。

# 来取ubuntu18.04基础镜像
docker pull ubuntu:18.04

# 重命名一下镜像 ==> 不做也行
docker tag ubuntu:18.04 ubuntu/18.04:v0

# 创建容器 ==> port9999和挂载目录workspace可以根据你的实际情况填写
docker run --name=atlas_infer -p 9999: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 /workspace:/workspace ubuntu/18.04:v0 /bin/bash

exit

# 重新进入容器
docker start atlas_infer
docker attach atlas_infer

这样基础容器已经创建完成,下面进行容器开发环境的构建。

# 首先确保你在容器中能够正常调用加速卡驱动
npu-smi info 
# 如果没有找到npu-smi,则在宿主机find一下,然后拷到容器放到容器的系统环境就行

然后进行cann-toolkit的安装。

创建 HwHiAiUser 用户,cann的一些安装依赖于 HwHiAiUser 用户。

# 创建 HwHiAiUser 用户组
groupadd HwHiAiUser

# 创建 HwHiAiUser 用户
useradd -g HwHiAiUser -r -m -s /bin/bash HwHiAiUser

这个时候你 ls /home 已经能够发现有了 HwHiAiUser 用户目录了。

昇腾的开发环境强依赖于 python3.7.5,所以我们得先装上它。安装 python3.7.5 一般有两种方式,即采用源码编译 和 采用anaconda的conda来创建python3.7.5的环境,然后软链python和pip可执行程序到系统环境。为了方便起见,这里采用anaconda的方式,先安装上 Anaconda3,然后用 conda 安装 python3.7.5 环境。这里安装 Anaconda3 比较简单,略过,附上不限速下载源:anaconda

# 使用conda管理python3.7.5环境
# 创建 python3.7.5 环境
conda create -n atlas_py375 python=3.7.5

# 安装 python 依赖
conda activate atlas_py375

pip install -i https://pypi.douban.com/simple attrs psutil decorator numpy protobuf==3.11.3 scipy sympy cffi grpcio grpcio-tools requests
            
# 安装sdk依赖
apt -y update
apt install -y gcc g++ make cmake zlib1g zlib1g-dev libsqlite3-dev openssl libssl-dev libffi-dev unzip pciutils net-tools dkms

然后开始安装sdk。

# 已经有驱动了,就不用再安装了
# 直接进行cann的安装

# python环境构建
# 软链python
ln -s /root/anaconda3/envs/atlas_py375/bin/python /usr/bin
ln -s /root/anaconda3/envs/atlas_py375/bin/python /usr/bin/python3.7
# 软链pip
ln -s /root/anaconda3/envs/atlas_py375/bin/pip /usr/bin
ln -s /rooot/anaconda3/envs/atlas_py375/bin/pip3 /usr/bin

# 安装CANN
chmod +x ./Ascend-cann-nnrt_6.0.0.alpha002_linux-x86_64.run
./Ascend-cann-nnrt_6.0.0.alpha002_linux-x86_64.run --install

chmod +x ./Ascend-cann-nnae_6.0.0.alpha002_linux-x86_64.run
./Ascend-cann-nnae_6.0.0.alpha002_linux-x86_64.run --install

chmod +x ./Ascend-cann-toolkit_6.0.0.alpha002_linux-x86_64.run
./Ascend-cann-toolkit_6.0.0.alpha002_linux-x86_64.run --install

这些安装好之后,配置一份 source_env.sh 激活环境的脚本,当然你也可以直接写进 ~/.bashrc/etc/profile

# source_env.sh
source /usr/local/Ascend/ascend-toolkit/set_env.sh

最后可以测试一下环境是否已经配置好,可以测试一下atc环境:

atc

昇腾Atlas Docker开发环境搭建教程

这样就大功告成了。

好了,以上分享了 从基础镜像开始构建昇腾atlas docker开发环境,希望我的分享能对你的学习有一点帮助。

 【公众号传送】

《极智开发 | 从基础镜像开始构建昇腾atlas docker开发环境》

畅享人工智能的科技魅力,让好玩的AI项目不难玩。邀请您加入我的知识星球, 星球内我精心整备了大量好玩的AI项目,皆以工程源码形式开放使用,涵盖人脸、检测、分割、多模态、AIGC、自动驾驶、工业等。不敢说会对你学习有所帮助,但一定非常好玩,并持续更新更加有趣的项目。 t.zsxq.com/0aiNxERDq

昇腾Atlas Docker开发环境搭建教程

本网站的内容主要来自互联网上的各种资源,仅供参考和信息分享之用,不代表本网站拥有相关版权或知识产权。如您认为内容侵犯您的权益,请联系我们,我们将尽快采取行动,包括删除或更正。
AI教程

深度学习编程入门 | 手动推导卷积层与池化层的计算过程

2023-12-14 5:45:14

AI教程

逻辑斯蒂回归模型多分类任务的案例实现

2023-12-14 7:45:14

个人中心
购物车
优惠劵
今日签到
有新私信 私信列表
搜索