学习深度学习用到图神经网络,配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就没问题了。
完成以上步骤就完全安装好了,结合文中参考两位同学的教程,加上我自己在安装过程中的一些经验,祝大家安装顺利~