YOLOv5 环境配置问题与解决

1.OSError: [WinError 1455] 页面文件太小,无法完成操作。
调大页面文件的大小 + 更改一下batch_size(彻底解决问题),详https://blog.csdn.net/weixin_43959833/article/details/116669523,为D盘分配虚拟内存,适当调大一些。

2.yolov5配置环境
Pytorch1.9+cuda10.2+cudnn7.0
具体:
cuda:cuda_10.2.89_441.22_win10.exe
Cudnn:cudnn-10.2-windows10-x64-v7.6.5.32.zip
Cudnn7.xxx具体看官网下载

不过到现在(202.7.25)for cuda10.2的cudnn版本已经更新到V8.2.1,如下图,还没有尝试过,不知都能不能用。

官网均可下到,下载cudnn需要有账号
尝试过:
Pytorch1.9+cuda11.1+cudnn8.0

Cudnn v8.0.5
这是我第一次配置yolov5环境是的搭配,我的nvidia显卡对应的cuda为11.1
(win+R 输入cmd 进入终端 输入nvidia-smi即可查看)

配置完成后,训练数据集时出现了问题

所有的loss指标,box loss,cls loss, obj loss,都为nan,map均为0,
查阅资料说时学习率太高,但是经过查看后学习率为lr0=0.01,lr1=0.2,正常的。后来把cuda11.1换为cuda10.2就解决了问题。
3.CUDA out of memory.(已解决)
详见:https://blog.csdn.net/weixin_43398590/article/details/105383173?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522162725880516780274133152%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257D&request_id=162725880516780274133152&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2allfirst_rank_v2~rank_v29-1-105383173.pc_search_result_cache&utm_term=cuda+out+of+memory&spm=1018.2226.3001.4187

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值