在运行Yolov5目标检测模型时遇到CUDA设备类型张量无法转换为numpy的错误。解决方法是在代码中添加'.cpu()'，将张量复制到主机内存。此问题出现在Windows系统的anaconda环境中，路径为'anaconda-envs-虚拟环境名称-Lib-site-packages-torch'，而在Linux系统中，路径位于'miniconda'目录下。确保修改正确路径以解决问题。
在运行yolov5目标检测代码时报错如下：
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
解决办法：在此处添加".cpu()"即可
win系统这个部分代码的路径是 ：
anaconda-envs-虚拟环境名称-Lib-site-packages-torch
linux系统代码的路径在：root目录下的miniconda，不要搞错了
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
