dgl学习笔记05-在GPU上使用DGL图

这个很简单，跟使用pytorch的GPU计算是一样的，就不解释了，直接上代码

代码：

import dgl
import torchdevice = ''
if torch.cuda.is_available():device = 'cuda:0'
else:device = 'cpu'u, v = torch.tensor([0, 1, 2]).to(device), torch.tensor([2, 3, 4]).to(device)
graph = dgl.graph((u, v))
nodes_feaure = torch.randn((graph.num_nodes(), 5)).to(device)
edges_feature = torch.randn((graph.num_edges(), 3)).to(device)
graph.ndata['x'] = nodes_feaure
graph.edata['e'] = edges_feature
print(graph)

dgl学习笔记05-在GPU上使用DGL图相关推荐

MySQL学习笔记05【多表操作、三大范式、数据库的备份和还原】
MySQL 文档-黑马程序员(腾讯微云):https://share.weiyun.com/RaCdIwas 1-MySQL基础.pdf.2-MySQL约束与设计.pdf.3-MySQL多表查询与事务 ...
opencv学习笔记05
原创:opencv学习笔记05 OpenCV-Python教程:40.ORB https://www.jianshu.com/p/49a84ddef11d ORB最重要的事情是它是OpenCV实验室出 ...
嵌入式算法移植优化学习笔记5——CPU,GPU,TPU,NPU都是什么
嵌入式算法移植优化学习笔记5--CPU,GPU,TPU,NPU都是什么一.什么是CPU? 二.什么是GPU? 三.什么是NPU? 四.什么是TPU? 附: 随着AI的广泛应用,深度学习已成为当前AI ...
【JVM 学习笔记 05】：JVM性能调优工具的使用和优化案例
[JVM 学习笔记 05]:JVM性能调优工具的使用 1. 使用 jstat(命令行工具) 查看线上系统的JVM运行状况 1.1 常用命令 1.2 使用技巧 1.2.1 随着系统运行,每秒钟会在年轻代 ...
【计算机网络学习笔记05】典型企业网络架构、传输介质
[计算机网络学习笔记05]典型企业网络架构.传输介质企业网络是 Internet 的重要组成部分,随着企业业务需求而不断地变化.为了了解企业网络如何满足企业业务需求,必须了解典型的企业网络架构以及网 ...
翻转电平函数实现LED闪烁-STM32电控学习笔记05
翻转电平函数实现LED闪烁-STM32电控学习笔记05 day5:2022/9/23 [函数介绍] 在前面帖子大致了解了一下HAL_GPIO_WritePin()函数和HAL_Delay()函数的用法 ...
Vue学习笔记05 组件的自定义事件-组件通信-$nextTick-脚手架解决ajax跨域-插槽-过渡动画
文章目录 Vue学习笔记05 父组件给子组件传值注意点子组件给父组件传值父组件接受子组件的传值通过函数组件的自定义事件事件绑定的第一种写法 @或v-on 事件绑定的第二种写法:使用ref ...
ESP32 单片机学习笔记 - 05 - AP/Smart Config
ESP32 单片机学习笔记 - 05 - AP/Smart Config 终于把感觉必要的基础外设学完了,开始学esp32的主要特色功能--物联网~~?(大概) 一.WIFI热点 AP模式编程指南: ...
JavaWeb黑马旅游网-学习笔记05【分类数据展示功能】
Java后端学习路线笔记汇总表[黑马程序员] JavaWeb黑马旅游网-学习笔记01[准备工作] JavaWeb黑马旅游网-学习笔记02[注册功能] JavaWeb黑马旅游网-学习笔记03[登陆和 ...
JavaWeb-综合案例(用户信息)-学习笔记05【分页查询功能】
Java后端学习路线笔记汇总表[黑马程序员] JavaWeb-综合案例(用户信息)-学习笔记01[列表查询] JavaWeb-综合案例(用户信息)-学习笔记02[登录功能] JavaWeb-综合案 ...

dgl学习笔记05-在GPU上使用DGL图

dgl学习笔记05-在GPU上使用DGL图相关推荐

最新文章

热门文章