昇腾虚拟机固件与驱动安装
作者:@wangxiyuan
华为云目前提供了基于Ascend 310的虚拟机,实例类型为Ai1,支持的OS为ubuntu 18.04和centos 7.6。本文以ubuntu 18.04的4U16G并搭载两块ascend 310的VM为例,说明如何初始化开发环境。
卸载旧版固件
华为云的Ai1实例默认已经安装了Ascend的驱动和lib库,但版本可能已过时,如果想使用新版的开发环境,则需要先删除旧版固件。
1
2
3进入/usr/local/Ascend,
cd /usr/local/Ascend
按照opp、ascend-tookit、driver的顺序,依次进入相关目录的script目录中,执行uninstall.sh文件卸载完成后,需要重启VM
安装驱动和开发库
从昇腾官方下载最新的驱动和开发包
- CANN-NNRT(NNRT是推理运行时包,开发环境可以不用安装,已被toolkit集成)
- CANN-TOOLKIT
- Ascend 310 driver
下载对应的
.run
包,拷贝到VM中,赋予执行权限,按照driver、nnrt、toolkit的顺序依此安装。1
2
3
4
5
6
7
8
9
10
11
12
13
14安装driver
./A300-3010-npu-driver_21.0.4_linux-x86_64.run --full
检查安装结果,显示两块OK的ascend 310卡
npu-smi info
安装nnrt
./Ascend-cann-nnrt_5.1.RC2.alpha002_linux-x86_64.run --install:
安装依赖
apt-get install -y gcc g++ make cmake zlib1g zlib1g-dev openssl libsqlite3-dev libssl-dev libffi-dev unzip pciutils net-tools libblas-dev gfortran libblas3 libopenblas-dev
升级pip
pip3 install --upgrade pip
安装pip依赖
pip3 install attrs numpy decorator sympy cffi pyyaml pathlib2 psutil protobuf scipy requests
安装toolkit
./Ascend-cann-toolkit_5.1.RC2.alpha002_linux-x86_64.run --install测试
按照官方指导,下载example工程并编译执行即可。