极智开发 | 从基础镜像开始构建昇腾atlas docker开发环境
欢迎存眷我的公众号 [极智视界],获取我的更多经验分享大师好,我是极智视界,本文介绍一下 从基础镜像开始构建昇腾atlas docker开发环境。
邀您插手我的常识星球「极智视界」,星球内有超多好玩的项目实战源码下载,链接:https://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开发环境,但愿我的分享能对你的学习有一点辅佐。
<hr/>【公众号传送】
畅享人工智能的科技魅力,让好玩的AI项目不难玩。邀请您插手我的常识星球,星球内我精心整备了大量好玩的AI项目,皆以工程源码形式开放使用,涵盖人脸、检测、分割、多模态、AIGC、自动驾驶、工业等。不敢说会对你学习有所辅佐,但必然非常好玩,并持续更新更加有趣的项目。https://t.zsxq.com/0aiNxERDq
页:
[1]