解决“TypeError: can‘t convert cuda:0 device type tensor to numpy. ......”问题

最新推荐文章于 2025-05-13 10:56:51 发布

Zpadger

最新推荐文章于 2025-05-13 10:56:51 发布

阅读量9.2w

点赞数 89

CC 4.0 BY-SA版权

分类专栏： Python debug 文章标签： python 深度学习 numpy pytorch

本文链接：https://blog.youkuaiyun.com/SpadgerZ/article/details/115584829

在使用Pytorch训练模型时遇到TypeError，原因是尝试将CUDA tensor直接转换为numpy数组。解决方法是先将CUDA tensor通过`.cpu()`迁移到CPU，然后再调用`.numpy()`进行转换。此问题在Python 3.7环境中出现，但在Python 3.8环境中未复现，可能与Python版本有关。

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

问题描述

在训练Pytorch模型的时候，报错

Traceback (most recent call last):
  File "train.py", line 469, in <module>
    train(hyp, tb_writer, opt, device)
  File "train.py", line 347, in train
    save_dir=log_dir)
  File "/home/xxx/Detection/test.py", line 176, in test
    plot_images(img, output_to_target(output, width, height), paths, str(f), names)  # predictions
  File "/home/xxx/Detection/utils/utils.py", line 914, in output_to_target
    return np.array(targets)
  File "/home/xxx/anaconda3/envs/pytorch1.5/lib/python3.7/site-packages/torch/tensor.py", line 492, in __array__
    return self.numpy()
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.