用gpu跑代码时出现的问题

1、当代码都调试后，使用gpu跑，出现：
can't convert cuda:0 device type tensor to numpy.Use Tensor.cpu to copy the tensor to host memory first
此时根据提示，找到代码处的loss.detach().将其该成loss.detach().cpu().numpy()即可，原因：gpu式的tensor不能直接转成numpy

待更新！！！

posted @   原来是只呆燕  阅读(196)  评论(0)  编辑  收藏  举报
相关博文：
· 一些代码的实验
· model.to(device)依然说权重不是cuda类型的
· 训练神经网络时报错：can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
· TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
· RuntimeError: Can't call numpy() on Tensor that requires grad. Use tensor.detach().numpy() instead.
阅读排行：
· 被坑几百块钱后，我竟然真的恢复了删除的微信聊天记录！
· 没有Manus邀请码？试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单，K8s 太复杂？w7panel 让容器管理更轻松！
点击右上角即可分享
微信分享提示