dgl的cuda版本安装+对应torch的cuda版本安装

优秀教程

1、pytorch 1.8.1 + CUDA11.1 对应的DGL版本是0.6.1（linux、windows也适用）
2、CUDA+pytorch+DGL安装
3、安装dgl-cuda
4、torch 安装备忘录

dgl文件下载

https://anaconda.org/dglteam/repo
1

可以将下面网址中的cu111改成你想要的版本，比如cu113、cu117等

https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu111/
1

pytorch文件下载

https://download.pytorch.org/whl/torch_stable.html
1

PyTorch中torch、torchvision、torchaudio版本对应关系

https://blog.csdn.net/AdamCY888/article/details/131021747

https://blog.csdn.net/shiwanghualuo/article/details/122860521
1
2
3

pytorch 1.8.1 + CUDA11.1 对应的DGL-cu111版本是0.6.1

我的环境

显卡: RTX 3060
cuda: cu111
python: 3.7
dgl-cu111: 0.6.1
torch: 1.8.1+cu111
1
2
3
4
5

安装结果

调用程序