图神经网络框架DGL使用记录_dgl-cu102和dgl-CSDN博客文章浏览阅读3.2k次,点赞4次,收藏4次。最近跟着导师学图神经网络,开始看论文搭模型跑代码,用的DGL框架,一开始不舒服被坑的有点惨,特此记录一下使用情况,避免其他学习的同学踩坑。一、关于安装一定先安装pytorch再安装dgl,因为dgl是依赖于pytorch,也不一定说你颠倒安装顺序就一定不对,但是可能容易出现一些包找不到的问题。二、关于cpu和gpupip install dgl这样默认安装的是cpu版本,没法调用GPU运算,安装GPU版本需要pip install dgl-cu102pip install dgl-cu100_dgl-cu102和dglhttps://blog.csdn.net/weixin_43918046/article/details/122181896安装cuda版本的dgl库_pytorch_哇咔君i-华为云开发者联盟选择离线安装吧,更快更安全进入https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu102/。最好下载高版本的,我之前下的0.4版本的,dgl没有load_graph()模块,GATConv()网络中也没有allow_zero_in_degree这个参数。。所以重新下载0.6版本的dgl,完美解决这些问题。。注意和自己的cuda版本对应上,cu101表示cu 哇咔君i 华为云开发者联盟
https://huaweicloud.csdn.net/63808c4fdacf622b8df8a7e3.htmlLinks for dgl-cu102
https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu102/
dgl的cuda版本
最新推荐文章于 2024-06-19 10:41:06 发布