weixin_43916559的博客

!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都