yolov5训练太慢的解决方案
yolov5
yolov5 - Ultralytics YOLOv8的前身,是一个用于目标检测、图像分割和图像分类任务的先进模型。
项目地址:https://gitcode.com/gh_mirrors/yo/yolov5
免费下载资源
·
问题原因
训练太慢大多是因为没有安装CUDA和pytorch,导致的只有cpu在跑,显卡没跑
这就是很典型的。
解决方案
第一步:安装CUDA
在本机上面安装CUDA,记住只有N卡可以安装,一开始的电脑是自带CUDA的。
如果不是自带的CUDA,下载落后一点的,可能打游戏的时候会有点毛病。
nvcc -V
本机安装成功后,其他虚拟环境也是有的。
至于安装CUDA的教程,我就不教了,网上一大片都有。
第二步:安装pytorch
官网:Previous PyTorch Versions | PyTorch
根据CUDA的版本挑选合适的。
就比如说我这个,是10.2 版本的。
就这个了。
兄弟们,中计了,这个是默认CPU的
这个才是GPU的,不过网速巨慢
GitHub 加速计划 / yo / yolov5
49.28 K
16.02 K
下载
yolov5 - Ultralytics YOLOv8的前身,是一个用于目标检测、图像分割和图像分类任务的先进模型。
最近提交(Master分支:1 个月前 )
f7322921
* Delete greetings.yml
* Update format.yml
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> 4 天前
12b577c8
* Refactor code for speed and clarity
* Auto-format by https://ultralytics.com/actions
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Update README.zh-CN.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Auto-format by https://ultralytics.com/actions
* Update README.md
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
* Auto-format by https://ultralytics.com/actions
---------
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> 10 天前
更多推荐
已为社区贡献10条内容
所有评论(0)