Windows安装yolov8无法使用cuda问题解决办法

发布时间:2024年01月18日

18年的mbp已经训练不动了,因此转到Windows上来使用yolov8。一开始装上后使用Win+G显示性能状态发现GPU根本就没用起来,还是在CPU上跑。为了能到GPU上跑起来折腾了一天才好,记录一下踩坑过程。

要使用yolov8必须要装pytorch。但是yolov8的pip安装指令会自动安装一个pytorch版本,就又导致和cuda版本对不上号一直用不起来gpu。

所以必要条件:pytorch和cuda必须要对上型号。详细踩坑过程就不说了,pytorch对应cuda11.8、12.3、12.1几个版本来回装了几遍。说说正确运行的解决办法:
1.显卡驱动更新到最新
2.走官网安装pytorch对应cuda11.8版本的
3.走cuda官网找到11.8版本并安装
4.在cmd里使用conda安装yolov8,这一步很重要,pip装的不行
5.确认cuda的环境变量都对的(默认应该都对的,不对的话问问GPT怎么设置)
6.在训练的时候要设置workers=0如model.train(data=‘./data.yaml’, epochs=3, workers=0),GPT的解释是workers=0使用了单线程数据加载,如果不用这个会卡在加载数据那。

最好每个东西安装好了根据GPT提示用对应的代码检查一下是否安装正确。

如果还不行,就要使用代码指定使用GPU:

from ultralytics import YOLO
import torch
device = torch.device("cuda")

# 加载预训练的YOLO模型(推荐用于训练)
model = YOLO('./yolov8l.pt')
model.to(device)
results = model.train(data='./data.yaml', epochs=3, workers=0)
文章来源:https://blog.csdn.net/eevee_1/article/details/135674175
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。