当前位置:   article > 正文

yolov5训练太慢的解决方案_yolo用gpu训练怎么超级慢

yolo用gpu训练怎么超级慢

问题原因

训练太慢大多是因为没有安装CUDA和pytorch,导致的只有cpu在跑,显卡没跑

这就是很典型的。

解决方案

第一步:安装CUDA

在本机上面安装CUDA,记住只有N卡可以安装,一开始的电脑是自带CUDA的。

如果不是自带的CUDA,下载落后一点的,可能打游戏的时候会有点毛病。

nvcc -V

本机安装成功后,其他虚拟环境也是有的。

至于安装CUDA的教程,我就不教了,网上一大片都有。

第二步:安装pytorch

官网:Previous PyTorch Versions | PyTorch

根据CUDA的版本挑选合适的。

 就比如说我这个,是10.2 版本的。

就这个了。

兄弟们,中计了,这个是默认CPU

这个才是GPU的,不过网速巨慢

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家小花儿/article/detail/452378
推荐阅读
相关标签
  

闽ICP备14008679号