TypeError: can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor
最新推荐文章于 2024-05-08 08:49:50 发布
在PyTorch里,直接从CUDA取数据时,使用output.cuda().data.numpy()会报错“TypeError: can’t convert CUDA tensor to numpy”。解决办法是将代码改成output.cuda().data.cpu().numpy()。



1万+

被折叠的 条评论
为什么被折叠?



