dgl cuda版本安装

dgl cuda版本安装

https://www.dgl.ai/pages/start.html
链接: dgl官网安装 相信如果按照这个成功了 也不需要看这个了吧。 针对最简单的测试：需要按照cuda版本的dgl，如何按照？非常容易

import dgl
import torch as th
u, v = th.tensor([0, 1, 2]), th.tensor([2, 3, 4])
g = dgl.graph((u, v))
g.ndata['x'] = th.randn(5, 3)   # 原始特征在CPU上
cuda_g = g.to('cuda:0')         # 接受来自后端框架的任何设备对象
print(cuda_g.device)

dgl-whl 下载地址 下载到本地是最容易的。
找到对应的版本，我的cuda 116 python 37 win64

然后打开虚拟环境，到这个文件夹位置，使用pip install 按照这个whl就可以成功了。
如
下面展示一些 内联代码片。

pip install "dgl-1.1.2+cu116-cp38-cp38-win_amd64.whl"

最后的结果：