Linux安装PyTorch和DGL的GPU版本

背景

复现一篇论文，要求指定pytorch>=1.8.1，dgl==0.4.0（后续发现0.4.0~0.4.3的都可以，高版本不支持是因为API包变了，其实自己小小改一下代码高版本也能用，但是我懒不想折腾）

安装步骤

首先检查你的CUDA版本，然后去选择低于或等于你版本的。比如我的是12.3，实测10.2版本的一样能用，因为是向下兼容的，所以不用太纠结版本一致。

• pytorch历史版本【官网】：https://pytorch.org/get-started/previous-versions/
  如果是whl安装要安装好几个whl，自行搜索这方面的博客。

• dgl历史版本【清华源pip】：https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu102/，cu版本的只有90, 100，101，102, 110，111，挑一个最接近你cuda版本的（向下兼容），直接pip install xxx.whl

• dgl历史版本【anaconda源】：https://conda.anaconda.org/dglteam/linux-64，最不全的资源，这个下载下来好像要挪文件，不清楚，自行搜索

• dgl历史版本【未知】：https://data.dgl.ai/wheels/repo.html，不知道这个whl是哪方的，和清华源的互补

补充

一开始我是想着安装高一点的dgl版本，2.1.x版本文档我都看完了，也知道老代码该怎么改了。因为我显卡的cuda版本12.3，看到dgl官方说支持12.1了，我就屁颠屁颠去搞了。pytorch安装的是2.1.1，dgl安装的是2.1.0.cu121，python3.8，结果测试安装是否成功时torch一切正常，但是import dgl就报错，说什么torchdata这个模块找不到。搜了搜没啥有用的信息，我选择不折腾，最后安装的是pytorch = 1.10.1(CUDA 10.2)，dgl = cu102-0.4.2，python = 3.7，完事了。