can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory

最新推荐文章于 2024-03-01 14:46:48 发布

FakeOccupational

最新推荐文章于 2024-03-01 14:46:48 发布

阅读量1.2k

点赞数

分类专栏：语言学习笔记文章标签： python

本文链接：https://blog.csdn.net/ResumeProject/article/details/121643345

版权

183 篇文章 4 订阅

订阅专栏

该博客探讨了如何将cuda设备上的numpy数组复制到主机内存，以便进行cpu.unmpy()操作，解决'can't convert cuda:0 device type tensor to numpy'错误。重点在于使用Tensor.cpu()方法实现数据迁移技巧。

摘要由CSDN通过智能技术生成

a.numpy()

改为

a.cpu().unmpy()

can't convert cuda:0 device type tensor to numpy. 
Use Tensor.cpu() to copy the tensor to host memory first.

关注

专栏目录