我试图在我的电脑中使用'Yolov5',我搜索了视频和文档等等......几乎尝试了我找到的所有解决方案,但没有进展......
尝试过 Tensorflow(甚至可能不需要)和 Tourch(主要需要这个)库,所有显示的东西都是 Cpu。
这是 yolov5 链接,在我的电脑中也可以在“Google Colab”中找到(但使用 Cpu)
https://github.com/ultralytics/yolov5
在 Windows 10 上工作,显卡“Rtx1660”,Anaconda v3(Jupyter 和 Pycharm),使用 python 3.7,Cuda v10.0.130,它是 Cudnn。
还安装了“Cuda”和“Cudnn”,它在 matlab2020b 和 2021a 上运行良好。
有没有可以使用 Gpu 与 Yolov5 一起训练的方法或程序?