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

在运行yolov5目标检测代码时报错如下:
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.

解决办法:在此处添加".cpu()"即可
yolov5 can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host_第1张图片
win系统这个部分代码的路径是 :
anaconda-envs-虚拟环境名称-Lib-site-packages-torch
linux系统代码的路径在:root目录下的miniconda,不要搞错了

你可能感兴趣的:(机器学习算法,cv,numpy,python,开发语言,计算机视觉,目标检测)