admin管理员组

文章数量:1650768

PyTorch 报错

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

很显然,我们没有把向量同步到我们模型的所用的计算设备上。

一般,我们用 <你的dgl对象名称>.to(<你的设备>) 来解决。如下。

g.to(torch.device('cuda:0'))

但是发现无效。

后来搜索到了相关的资料:

  • https://github/dmlc/dgl/issues/2171#issuecomment-691985268

如上,g.to(device) 仅仅是返回一个对象,并不改变其自身的属性。因此,我们要想改变 g &#

本文标签: DGLErrorassignbaseDGLffi