昇腾910平台安装驱动、固件、CANN toolkit、pytorch

本文使用的昇腾910平台操作系统是openEuler,之前没了解过,不过暂时感觉用起来和centOS差不多。系统架构是ARM,安装包基本都是带aarch64字样,注意和x86_64区别开,别下错了。

安装依赖

cmake

通过yum安装的cmake版本较老,需要源代码安装

wget https://cmake.org/files/v3.21/cmake-3.21.3.tar.gz
tar -zxvf  cmake-3.21.3.tar.gz
cd cmake-3.21.3
./bootstrap
gmake -j16
sudo gmake install

anaconda

使用系统自带的python3也可以,不过anaconda的虚拟环境我用习惯了,感觉很方便。
在https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/ 直接找最新的版本。我用的是https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-2023.09-0-Linux-aarch64.sh
安装完别忘记把环境变量加入.bashrc文件。

其它依赖

yum install -y gcc gcc-c++ make cmake unzip zlib-devel libffi-devel openssl-devel pciutils net-tools sqlite-devel lapack-devel gcc-gfortran

卸载旧的驱动、固件、CANN toolkit

卸载驱动

cd /usr/local/Ascend/driver/script
bash uninstall.sh

卸载固件

cd /usr/local/Ascend/firmware/script
bash uninstall.sh

卸载nnae

cd /usr/local/Ascend/nnae/6.3.RC1/script
bash uninstall.sh

卸载ascend-toolkit

cd /usr/local/Ascend/ascend-toolkit/
bash cann_uninstall.sh

/ascend-toolkit文件夹下可能安装了多个版本,用下面的命令找一下这些版本脚本位置,都卸载干净:

find . -type f -name "cann_uninstall.sh"

安装驱动和固件

下载链接 https://www.hiascend.com/hardware/firmware-drivers/community?product=4&model=10&cann=7.0.0.alpha003&driver=1.0.21.alpha
按你的平台和安装方式筛选:

在这里插入图片描述
我是下面这两个:
Ascend-hdk-910-npu-driver_23.0.rc3_linux-aarch64.run
Ascend-hdk-910-npu-firmware_7.0.0.5.242.run

以root用户登录,将驱动和固件包上传。创建驱动运行用户HwHiAiUser。

groupadd -g 1000 HwHiAiUser
useradd -g HwHiAiUser -u 1000 -d /home/HwHiAiUser -m HwHiAiUser -s /bin/bash

增加可执行权限:

chmod +x Ascend-hdk-910-npu-driver_23.0.rc3_linux-aarch64.run
chmod +x Ascend-hdk-910-npu-firmware_7.0.0.5.242.run

安装驱动:

./ Ascend-hdk-910-npu-driver_23.0.rc3_linux-aarch64.run --full --install-for-all

出现类似如下回显信息,说明安装成功。
Driver package installed successfully!

还可以通过执行npu-smi info命令检查驱动是否加载成功。

安装固件:

./ Ascend-hdk-910-npu-firmware_7.0.0.5.242.run --full

出现类似如下回显信息,说明安装成功。
Firmware package installed successfully! Reboot now or after driver installation for the installation/upgrade to take effect

驱动和固件安装完成后,重启系统。

reboot

安装CANN

下载安装包:https://www.hiascend.com/developer/download/community/result?module=cann&product=4&model=10
我是这个:
Ascend-cann-toolkit_7.0.0.alpha003_linux-aarch64.run

增加可执行权限:

chmod +x Ascend-cann-toolkit_7.0.0.alpha003_linux-aarch64.run

安装

./Ascend-cann-toolkit_7.0.0.alpha001_linux-x86_64.run --install

安装过程中输入Y同意协议。安装完成后,若显示如下信息,则说明软件安装成功。
[INFO] Ascend-cann-toolkit install success

配置CANN环境变量,将下面这句话加入.bashrc:

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

安装pytorch

创建虚拟环境

CANN对python版本有要求,同时需要安装一些python包,可以通过下面的脚本检查是否满足:

cd /usr/local/Ascend/ascend-toolkit/latest/aarch64-linux/bin
bash prereq_check.bash

我选用的版本是python3.8.11,通过下面命令创建虚拟环境:

conda create -n py38 python=3.8.11

激活环境

conda activate py38

安装依赖包

pip install pyyaml
pip install numpy
pip install attrs cloudpickle decorator psutil scipy synr==0.5.0 tornado absl-py
pip install google-cloud google-cloud-vision
protobuf==3.20.0

安装速度慢的话可以加上指定源,比如:

pip install -i https://mirrors.aliyun.com/pypi/simple/ pyyaml

注意protobuf的版本。

安装pytorch及adapter

参考 https://gitee.com/ascend/pytorch
我安装的是2.1.0版本,可以直接通过pip快速安装

pip3 install torch==2.1.0
pip3 install pyyaml
pip3 install setuptools
pip3 install torch-npu==2.1.0rc1

验证:
第一步:
python3 -c "import torch;import torch_npu;print(torch_npu.npu.is_available())"
返回True说明可以成功调用NPU。
第二步:

import torch
import torch_npux = torch.randn(2, 2).npu()
y = torch.randn(2, 2).npu()
z = x.mm(y)print(z)

能正常计算说明安装成功。可能遇到算子构造失败的情况,检查python版本是否符合要求,CANN安装之前是否卸载干净。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/307942.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

ES的使用(Elasticsearch)

ES的使用(Elasticsearch) es是什么? es是非关系型数据库,是分布式文档数据库,本质上是一个JSON 文本 为什么要用es? 搜索速度快,近乎是实时的存储、检索数据 怎么使用es? 1.下载es的包(环境要…

Word 将页面方向更改为横向或纵向

文章目录 更改整个文档的方向更改部分页面的方向方法1:方法2: 参考链接 更改整个文档的方向 选择“布局”>“方向”,选择“纵向”或“横向”。 更改部分页面的方向 需要达到下图结果: 方法1: 选:中你要在横向页面…

新品出击 | 软网关BLIoTLink免费发布

新品出击|软网关BLIoTLink免费发布 BLIoTLink是一款免费的物联网协议转换软件,可以部署在任何基于Linux OS的系统(Linux、Debian、Ubuntu、FreeRTOS、RT-Thread)中,使用灵活,可以实现数据的采集以及接入网络平台。 BL…

Mybatis行为配置之Ⅳ—日志

专栏精选 引入Mybatis Mybatis的快速入门 Mybatis的增删改查扩展功能说明 mapper映射的参数和结果 Mybatis复杂类型的结果映射 Mybatis基于注解的结果映射 Mybatis枚举类型处理和类型处理器 再谈动态SQL Mybatis配置入门 Mybatis行为配置之Ⅰ—缓存 Mybatis行为配置…

Mac Pycharm在Debug模式报编码(SyntaxError)错误

1. 错误信息: Traceback (most recent call last):File "/Library/Developer/CommandLineTools/Library/Frameworks/Python3.framework/Versions/3.9/lib/python3.9/tokenize.py", line 330, in find_cookieline_string line.decode(utf-8) UnicodeDeco…

鸿蒙HarmonyOS-图表应用

简介 随着移动应用的不断发展,数据可视化成为提高用户体验和数据交流的重要手段之一。在HarmonyOS应用开发中,一个强大而灵活的图表库是实现这一目标的关键。而MPChart就是这样一款图表库,它为开发者提供了丰富的功能和灵活性,使得…

Adobe 设计精髓:创新的用户体验 | 开源日报 No.130

adobe/react-spectrum Stars: 10.1k License: Apache-2.0 React Spectrum Libraries 是一系列的库和工具,旨在帮助开发者构建适应性强、可访问性好且稳健的用户体验。 核心优势: 提供全面的可访问性和行为支持,符合 WAI-ARIA 编写实践&…

小型内衣洗衣机什么牌子好?口碑好的小型洗衣机

想必大家都知道,我们的内衣裤、袜子这些衣物对卫生方面的要求是比较的高,毕竟是贴身的衣物,因此是要分开清洗的,而不能够跟我们其他的大件衣服一起放入到大型洗衣机里进行混洗,很多就选择了分开单独的手洗,…

K8s陈述式资源管理

命令行:kubectl命令行工具 优点:90%以上的场景都可以满足 对资源的增删改查比较方便,对改不是很友好。 缺点:命令比较冗长,复杂,难记 声明式: 看到声明式都是k8s当中的yml文件来实现资源管理…

python+django网上银行业务综合管理系统vue_bvj8b

本课题主要研究如何用信息化技术改善传统网上银行综合管理行业的经营和管理模式,简化网上银行综合管理的难度,根据管理实际业务需求,调研、分析和编写系统需求文档,设计编写符合银行需要的系统说明书,绘制数据库结构模…

计算机中的数据表示方法

原码表示法: 整数表示,范围: 小数表示: 补码表示可以将减法运算通过加法实现

解决VNC连接Ubuntu服务器打开终端出现闪退情况

服务器环境 阿里云ECS服务器 操作系统:Ubuntu 20.0.4 如何使用VNC连接阿里云ECS服务器 1.阿里云官方指导:通过VNC搭建Ubuntu 18.04和20.04图形界面 2.新手入门ECS——ubuntu 20.04安装图形化界面和本地VNC连接 问题描述 使用VNC连接上新申请阿里云服…