项目场景:
训练pytorch模型时报错
问题描述
TypeError:can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy...
原因分析:
如果想把CUDA tensor格式的数据改成numpy时,需要先将其转换成cpu float-tensor随后再转到numpy格式。 numpy不能读取CUDA tensor 需要将它转化为 CPU tensor
解决方案:
提示:将报错代码self.numpy()改为self.cpu().numpy()即可
AAE_x_hat = AAE_x_hat.detach().cpu().numpy().squeeze()
参考链接
提示:https://blog.csdn.net/SpadgerZ/article/details/115584829
本文介绍了在PyTorch中处理CUDA Tensor到numpy转换时遇到的TypeError,并提供了解决方案:使用self.cpu().numpy()代替直接self.numpy()。实例代码和CSDN链接帮助开发者解决模型训练过程中的转换问题。

1万+

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



