在执行深度学习或GPU计算时遇到TypeError，提示无法将CUDA设备类型的张量转换为NumPy数组。解决方法是使用Tensor的`.cpu()`方法先将其复制到CPU内存，然后再调用`.numpy()`。
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
a1 = a1.data.cpu().numpy()
TypeError: can‘t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
a1 = a1.data.cpu().numpy()
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
