新手入门:DGL在昇腾上的安装问题

简介: 本文介绍了在aarch64架构和Python 3.10环境下安装DGL(Deep Graph Library)的过程。首先通过`uname -a`确认硬件架构,接着使用`python --version`检查Python版本。为确保兼容性,从指定链接下载适合的whl包或通过pip安装dgl。过程中遇到了torchdata版本不兼容的问题,通过降级torchdata至0.7.1版本解决。此外,针对NPU芯片适配,重新安装了与CANN 8.0.RC2兼容的torch和torch_npu组件。最终成功导入dgl包并准备进行模型训练验证。

首先需要确定下自己的硬件,在terminal中输入指令

uname -a

返回结果

aarch64

接着判断当前环境的python版本,在terminal中输入指令

python --version

返回结果

Python 3.10.14

可以判断需要安装aarch64的版本,如果使用whl包安装dgl,以该链接(data.dgl.ai/wheels/cu118/repo.html)为例,需要安装指定Python3.10-aarch64版本的包


比较便捷的方式是直接通过pip来安装,指令如下

pip install dgl

注意安装完成后,terminal显示成功安装的dgl版本以及对应的配套安装的其他包的版本,信息如下

Successfully installed dgl-2.1.0 sympy-1.13.1 torch-2.5.1 torchdata-0.10.1

接下来先简单导入下dgl看是否安装成功,终端执行指令如下

python -c "import dgl"

发现报错信息如下


按住Ctrl健点击traceback信息,可以检查安装路径下的python文件,经检查发现,pip install自动安装的配套torchdata的版本为0.10.1,但是该版本做了改动,已经没有了datapipes属性,由此可以推断需要改动torchdata的版本。经尝试,当dgl版本为2.1时,对应的0.7.1版本的torchdata满足要求,终端执行指令如下

pip install torchdata==0.7.1

安装成功后,再次尝试导入dgl

python -c "import dgl"

新的报错信息如下


按照指示安装缺失的包

pip install pydantic

安装成功后,再次尝试导入dgl

python -c "import dgl"

根据如下报错信息,推断在NPU芯片上,torch2.5.1与dgl2.1版本无法正常导入,并且注意的是,因为是在昇腾上运行,所以在安装torch时需要考虑到能否充分调动NPU算力,因此应当优先考虑torch_npu组件能覆盖的torch版本 pytorch: Ascend Extension for PyTorch (gitee.com)


接下来重新装torch,需要先确定CANN版本(在笔者机器上装的是CANN 8.0.RC2),并根据对照关系安装

版本对照关系

由此可知,应安装pytorch与torch_npu组件的版本与指令如下

pip3 install torch==2.1.0
pip3 install torch-npu==2.1.0.post6

安装成功后,再次尝试导入dgl,发现能成功导入

导入dgl包成功

接下来,基于导入的dgl包进行一个简单的模型训练验证,将在下文中记载。

相关文章
|
机器学习/深度学习
Whisper 整体架构图
Whisper 整体架构图
1035 0
|
15天前
|
人工智能 Linux API
Magpie, 轻量本地AI网关!免费Token用不完!
Magpie 是开源AI编程模型调度网关,上线一月获1.8万Star。它通过本地高性能网关统一托管多平台API密钥、账号与免费额度,实现跨Agent的模型秒级切换、静默额度轮换与资源复用,显著降低碎片化管理成本。(239字)
Magpie, 轻量本地AI网关!免费Token用不完!
|
应用服务中间件 nginx Windows
音视频系列六:Windows搭建Nginx+rtmp推流服务器
在前面 阿里云服务器搭建Nginx+rtmp推流服务器 中,我们已经配置把阿里云的rtmp推流服务搭建好了,用的是PC软件OBS来进行推流到阿里云服务器转发然后本地拉流。Windows也是大同小异,现在是用Windows进行推流服务的搭建,本地ffmpeg命令行推流,本地ffplay拉流播放/VLC拉流播放。
2157 0
音视频系列六:Windows搭建Nginx+rtmp推流服务器
|
测试技术
哪里不对改哪里!全能图像编辑模型Qwen-Image-Edit来啦
Qwen-Image-Edit基于20B Qwen-Image模型,融合视觉语义与外观控制,支持中英文文字精准编辑、风格迁移、IP创作等多重功能,具备SOTA性能,助力低门槛、高精度图像编辑。
4793 23
|
消息中间件 缓存 JSON
亚马逊SP-API开发实战:商品数据获取与操作
本文介绍了亚马逊SP-API接入流程,包括开发者注册、OAuth2.0认证示例及核心商品接口的使用。涵盖商品信息查询、批量查询、限流规则与错误处理,并提供最佳实践建议,如使用AWS Lambda与SQS实现高效数据同步。
亚马逊SP-API开发实战:商品数据获取与操作
|
机器学习/深度学习 编解码 人工智能
超越Transformer,全面升级!MIT等华人团队发布通用时序TimeMixer++架构,8项任务全面领先
一支由麻省理工学院、香港科技大学(广州)、浙江大学和格里菲斯大学的华人研究团队,开发了名为TimeMixer++的时间序列分析模型。该模型在8项任务中超越现有技术,通过多尺度时间图像转换、双轴注意力机制和多尺度多分辨率混合等技术,实现了性能的显著提升。论文已发布于arXiv。
1286 84
|
存储 安全 API
跨境电商API的法律风险与合规措施:走向全球市场的必经之路
跨境电商API在连接电商平台、商家与支付物流等环节中至关重要,但其应用面临数据安全、隐私保护及跨境传输等法律风险。本文分析了相关风险,并提出加强数据加密、身份验证、合规协议签订及安全审计等措施,助力企业合法合规地走向全球市场。以阿里巴巴韩国事件为例,强调遵循当地法规的重要性。未来,合规管理将成为企业全球化的核心竞争力。
|
数据采集
正则表达式匹配html标签,获取标签内容
正则表达式匹配html标签,获取标签内容
|
机器学习/深度学习 存储 安全
基于YOLOv8深度学习的舰船目标分类检测系统【python源码+Pyqt5界面+数据集+训练代码】目标检测、深度学习实战
基于YOLOv8深度学习的舰船目标分类检测系统【python源码+Pyqt5界面+数据集+训练代码】目标检测、深度学习实战

热门文章

最新文章