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"

在这里插入图片描述

最后的结果:
在这里插入图片描述

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐