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 发布

还在摸鱼嘛

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

阅读量710

点赞数

CC 4.0 BY-SA版权

文章标签： numpy python pytorch matplotlib ipython

本文链接：https://blog.csdn.net/weixin_45825077/article/details/133624019

博客主要讲述在GPU环境下使用PyTorch时，遇到无法将cuda:0设备类型张量转换为Numpy的问题，即“can’t convert cuda:0 device type tensor to numpy”，通过在变量后添加特定操作，给出修改后的代码，成功解决该问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在GPU环境下使用pytorch出现:can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

原代码

import matplotlib.pyplot as plt

# 折线图
plt.plot(x, y, label='sin')
plt.plot(x, a + b*x + c*x**2+ d*x**3, label='me')
# 图例
plt.legend()
# 画图
plt.show()

出现这个问题，只需要在变量后加上.cpu().numpy()即可。

修改后的代码：

import matplotlib.pyplot as plt

# 折线图
plt.plot(x.cpu().numpy(), y.cpu().numpy(), label='sin')
plt.plot(x.cpu().numpy(), (a + b*x + c*x**2 + d*x**3).cpu().numpy(), label='me')
# 图例
plt.legend()
# 画图
plt.show()

在这里插入图片描述
成功解决