- 博客(1)
- 收藏
- 关注
原创 !t.is_cuda() INTERNAL ASSERT FAILED 问题解决
!t.is_cuda() INTERNAL ASSERT FAILED 问题解决在用pytorch写一个简单的点云GCN时,在torch.sparse.mm(A,B)这一行代码报错,提示RuntimeError: !t.is_cuda() INTERNAL ASSERT FAILED at /pytorch/aten/src/ATen/native/sparse/SparseTensorMath.cpp:591【原因】:A或B并不都在cuda上,两个矩阵在不同的device上【解决方法】把A,B都
2020-09-27 02:38:35 9218
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人