0

我试图在我的电脑中使用'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 一起训练的方法或程序?

4

0 回答 0