当前位置:   article > 正文

YOLOv3-Pytorch(ultralytics)训练代码报错TypeError: can’t convert cuda:0 device type tensor to numpy. Use Te

YOLOv3-Pytorch(ultralytics)训练代码报错TypeError: can’t convert cuda:0 device type tensor to numpy. Use Te

TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

具体报错如下图所示:
![在这里插入图片描述](https://img-blog.csdnimg.cn/c9d42fe54d1c4868992b7b575ee45cea.png?x-oss-process=image/watermark,type_ZHJvaWRzYW5zZmFsbGJhY2s,shadow_50,text_Q1NETiBAd2VpeGluXzQ0ODg5NTgz,size_20,color_FFFFFF,t_70,g_se,x_16
此时打开utils.py文件,修改为下图所示代码即可
在这里插入图片描述
成功!!
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小小林熬夜学编程/article/detail/558524
推荐阅读
相关标签
  

闽ICP备14008679号