yolov5双目识别输出坐标代码报错

Question

请问我在跑您的双目识别算法的时候按b站视频的第一步输出坐标但是报错了

可以告诉我如何解决吗，我只需要做到输出坐标就好了，可以接受有偿

Answer 1

引自强大而免费的微信小程序：皆我百晓生。如有帮助，还望采纳。

---

根据错误提示，可以看出Torch没有启用CUDA编译，而您的代码中使用了CUDA相关的函数。此错误可能是由于安装的Torch版本未启用CUDA导致的。

要解决这个问题，您可以尝试以下解决方案之一：

1. 确保您安装的Torch版本已经启用了CUDA支持。可以通过在Python终端中运行以下代码来验证：

   import torch
   print(torch.cuda.is_available())
   

   如果输出为False，则表示您的Torch版本未启用CUDA支持。您可以尝试重新安装Torch或查看Torch文档获取如何启用CUDA的指南。

2. 如果您不需要使用CUDA，可以修改代码，使用CPU上的计算替代CUDA计算。可以通过将torch.cuda替换为torch来实现。例如，将代码中的 t.cuda(device) 替换为 t。

   # 原始代码
   return self.apply(lambda t: t.cuda(device))
   
   # 修改后代码
   return self.apply(lambda t: t)
   

   但是请注意，这样可能会导致程序在CPU上运行速度较慢。

希望这些解决方案能够帮助您解决问题。如果问题仍存在，请提供更多的信息和代码片段，以便我们更好地理解和帮助解决问题。