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

本文讲述了作者在将工作从MacBookPro迁移到Windows平台,使用yolov8进行深度学习时遇到的GPU问题,包括如何确保pytorch和CUDA版本匹配,显卡驱动更新,以及在训练过程中设置正确的环境变量和多线程使用方法。
摘要由CSDN通过智能技术生成

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)
  • 9
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值