環(huán)境記錄：

神船Z8筆記本（3060 Laptop顯卡，6G顯存）、win10、vs2019、opencv3.4.14、cuda11.2、cudnn11.2、驅(qū)動461.40。

訓(xùn)練數(shù)據(jù)集：

TT100K

訓(xùn)練重要參數(shù)：

batch=64
subdivisions=64

width=608
height=608

訓(xùn)練命令：

darknet.exe detector train data\tt100k.data cfg\yolov4-tt100k.cfg yolov4.conv.137 -map

報(bào)錯信息：

Try to set subdivisions=64 in your cfg-file.
CUDA status Error: file: D:\work-ai\darknet\src\dark_cuda.c : cuda_make_array() : line: 492 : build time: Feb 23 2022 - 12:15:10

CUDA Error: out of memory
Darknet error location: D:\work-ai\darknet\src\dark_cuda.c, check_error, line #69
CUDA Error: out of memory: No error

診斷信息：

? 使用nvidia-smi檢查GPU顯存使用情況，發(fā)現(xiàn)程序報(bào)錯未退出時顯存已被使用5.39G

診斷結(jié)論：

? GPU顯存不足。

處理方法：

? ?經(jīng)過向“謝謝大家我愛小謝”請教，將width和height調(diào)整為512后，訓(xùn)練不再報(bào)錯，此時GPU顯存使用值為4.5G左右波動。

總結(jié)

以上是生活随笔為你收集整理的笔记本3060显卡使用yolo-v4训练报错显存不足处理的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。