CUDA11.4下的DGL下载

本文讲述了作者在实验室服务器上将CUDA版本从11.4更新到1.12,并确保兼容PyTorch1.12和DGL1.0.0的过程,分享了如何通过官网找到合适的DGLwheel包以避免自动更新PyTorch版本的问题。
摘要由CSDN通过智能技术生成

实验室服务器cuda更新为11.4

找的文献所需环境其中pytorch==1.11、dgl==0.9.0

最终配置的环境为python3.8、pytorch==1.12、dgl==1.0.0,也能跑代码

首先,尝试直接pip dgl==0.9.0以及从官网(Deep Graph Library (dgl.ai))上直接pip都会更新虚拟环境下的pytorch版本

求助善良可爱的同学后,有效下载方法:

先打开dgl官网

官网上没有cuda11.4对应指令,实际尝试下来,cuda11.4和cuda11.3相近,同时都可兼容pytorch1.12,

打开安装指令对应的网址data.dgl.ai/wheels/cu113/repo.html

可以看到cuda11.3对应最低版本为dgl1.0.0,我们所需0.9.0,1.0.0也能用

选择cu113为cuda11.3,cp38为python3.8,Linux64位版本,下载到本地,之后上传到服务器项目文件同目录下

然后在项目中进行pip下载即可

pip dgl-1.0.0+cu113-cp38-cp38-manylinux1_x86_64.whl

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值