深度学习图神经网络,Windows系统下载CUDA、cuDNN、Torch、dgl

  学习深度学习用到图神经网络,配cuda环境费了一番功夫,把经验分享出来供大家讨论学习。

  在下载之前就要查询好后续需要安装的包兼容的CUDA和Torch等等,像dgl包在2024.6开始不再更新适配高版本的cuda和torch,所以要提前选好要下载的一系列东西的版本(不提前看好就会像我一样反反复复卸载下载好多次hh)

一、下载cuda,cudnn、torch

参考https://blog.csdn.net/yxn4065/article/details/139469418?spm=1001.2014.3001.5506

这里我推荐cuda、cudnn、torch都去官网下载,适配的版本很清晰可以找到,我自己刚开始用whl安装包下载后面弄得很乱,不好查询是否兼容。版本的兼容对于配置gpu至关重要!!!

下载pytorch时可以在这里选择以往的版本下载,最新版本的pytorch不适配dgl

以及这里下载cudnn的时候,我是分别把bin、include、lib复制到cuda同名的文件夹,此举参考了NVIDIA、CUDA、CUDNN、PyTorch安装吐血整理!!!_有nvidia还安装cuda吗-CSDN博客

二、下载dgl包

进入dgl官网,选择适配版本的dgl,我选择的是pyTorch2.2+CUDA12.1,conda包

在这里我推荐用conda安装,conda安装的优先级是要高于pip优先级的,我之前用pip装dgl,pip list显示已装好但import dgl又会报错,换成conda install就没问题了。

完成以上步骤就完全安装好了,结合文中参考两位同学的教程,加上我自己在安装过程中的一些经验,祝大家安装顺利~

  • 8
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值