当前位置 : 主页 > 编程语言 > python >

dgl._ffi.base.DGLError: Cannot assign node feature “feat“ on device cuda:0 to a graph on device c

来源:互联网 收集:自由互联 发布时间:2022-06-18
​​dgl/migrate-guide-0.5.md at master · dmlc/dgl · GitHub​​ 这个 链接 是一些dgl的版本变化可能会带来的问题以及解决方法 我的代码在这一行报错, 当我写了bg.to(args['device'])之后,还是一样的


​​dgl/migrate-guide-0.5.md at master · dmlc/dgl · GitHub​​

这个链接是一些dgl的版本变化可能会带来的问题以及解决方法

dgl._ffi.base.DGLError: Cannot assign node feature “feat“ on device cuda:0 to a graph on device cpu._pytorch

我的代码在这一行报错,

当我写了bg.to(args['device'])之后,还是一样的,我觉得很是奇怪。。。

我明明就是用的DGLGraph.to()。。。

那么后来我发现了问题,有没有那么一种可能就是,

需要bg = bg.to(args['device'])。。。

pass


上一篇:使用python实现smurf攻击
下一篇:没有了
网友评论