优秀教程
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
可以将下面网址中的cu111
改成你想要的版本,比如cu113
、cu117
等
https://pypi.tuna.tsinghua.edu.cn/simple/dgl-cu111/
pytorch文件下载
https://download.pytorch.org/whl/torch_stable.html
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
我的环境
显卡: RTX 3060
cuda: cu111
python: 3.7
dgl-cu111: 0.6.1
torch: 1.8.1+cu111