如何安装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

  

 

posted @ 2024-07-17 15:40  TW-NLP  阅读(12)  评论(0编辑  收藏  举报