优秀教程

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

dgl的cuda版本安装+对应torch的cuda版本安装_.net

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

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

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

pytorch文件下载

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

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

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

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

https://blog.csdn.net/shiwanghualuo/article/details/122860521

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

dgl的cuda版本安装+对应torch的cuda版本安装_.net_04


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

我的环境

显卡: RTX 3060
cuda: cu111
python: 3.7
dgl-cu111: 0.6.1
torch: 1.8.1+cu111

安装结果

dgl的cuda版本安装+对应torch的cuda版本安装_文件下载_06

dgl的cuda版本安装+对应torch的cuda版本安装_文件下载_07

调用程序

dgl的cuda版本安装+对应torch的cuda版本安装_文件下载_08