pytorch 错误:无法将cuda:0设备类型Tensor转换为numpy,请先使用Tensor.cpu()将Tensor复制到主机内存

gorkyyrv  于 2022-11-01  发布在  其他
关注(0)|答案(1)|浏览(275)

我正在PyTorch中尝试做一个MLP分类器。错误是从最后一个块的代码中产生的。我不知道为什么numpy会参与到这件事中来,有人能给我指出正确的方向吗?
第一个

oxcyiej7

oxcyiej71#

您正在使用的绘图函数plt.plot适用于numpy数组,而不适用于torch.tensor。因此,accHistory将被转换为numpy数组,并在此处失败。
有关详细信息,请参阅this answer

相关问题