MVGRL 代码复现 以及 DGL 安装问题

本文介绍了在复现MVGRL模型时遇到的DGL库安装问题,详细讨论了如何解决DGL CUDA版本不匹配的问题,以及在Cora、Citeseer和Pubmed数据集上的运行情况。通过离线安装DGL解决了pip安装慢的问题,并针对GPU内存不足进行了调整。实验结果显示MVGRL在节点分类任务中表现出有效性,但GPU资源消耗较大。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

原作者的code DGL数据集导入就有问题,其采用了 dgl 0.4.1,这里采用了zhrMVGRL的DGL版本,修复了cora数据集的问题,cora v2。

在这里插入图片描述
这里直接 pip install 出现问题,能下载下来,但是不能用,百度之后 应该是 pip install dgl默认安装cpu版本,需要到官网下载对应cuda版本的包,DGL官网
在这里插入图片描述
这里采用pip 安装,之后速度特别慢
在这里插入图片描述
尝试 conda 安装 更慢,,,
在这里插入图片描述

通过网址 直接自己下载whl ,再进行pip离线安装。 whl放到服务器 ,pip通过路径进行安装(前提:在这个虚拟环境里面)

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值