如何安装Ascend深度学习套件
1、 驱动安装
1.1 驱动测试
输入测试命令:
npu-smi info
结果如下:
1.2 Ascend驱动未安装
请参考Ascend驱动的安装文档,进行安装对应显卡的驱动,文档链接如下:https://support.huawei.com/enterprise/zh/doc/EDOC1100368783/2645a51f
2、 Ascend 深度学习套件的安装
2.1 若已经安装套件
(1)若安装了对应的Ascend-cann-toolkit和kernels,请跳过此步骤。
2.2 若没有安装上述套件
(2)若没有下载深度学习套件,请根据自己的显卡型号,进行下载,下面以Ascend 310P显卡为例,进入下面的下载地址:https://www.hiascend.com/developer/download/community/result?module=cann&cann=8.0.RC1.alpha001
2.2.1 下载文件
Ascend-cann-toolkit_8.0.RC1.alpha001_linux-aarch64.run Ascend-cann-nnae_8.0.RC1.alpha001_linux-aarch64.run Ascend-cann-kernels-310p_8.0.RC1.alpha001_linux.run
2.2.2 运行如下命令
chmod +x Ascend-cann-toolkit_8.0.RC1.alpha001_linux-aarch64.run ./Ascend-cann-toolkit_8.0.RC1.alpha001_linux-aarch64.run --check ./Ascend-cann-toolkit_8.0.RC1.alpha001_linux-aarch64.run –install
chmod +x Ascend-cann-kernels-310p_8.0.RC1.alpha001_linux.run ./Ascend-cann-kernels-310p_8.0.RC1.alpha001_linux.run --check ./Ascend-cann-kernels-310p_8.0.RC1.alpha001_linux.run --install
2.2.3 激活环境变量
source /usr/local/Ascend/nnae/set_env.sh source /usr/local/Ascend/ascend-toolkit/set_env.sh