Open chenxyyy opened 3 years ago
你可以试试减小batchsize 或者减小imgsize,yolov4就是比较吃显存
单张T4显卡训练时候,batchsize 的大小设置为10是没问题的。 4张显卡设置成16就不行了,感觉不是batchsize 的问题啊
@chenxyyy did you solve that issue. I am pruning a model, I tried different pruning thresholds and even reduce my batch size to 1 but the same error is comming.
RuntimeError: CUDA out of memory. Tried to allocate 170.00 MiB (GPU 0; 7.79 GiB total capacity; 5.54 GiB already allocated; 43.25 MiB free; 6.11 GiB reserved in total by PyTorch)
@SpursLipu can you suggest something. Also, I tried with pruning threshold between 0.5 to 0.01 but every time it's showing after pruning model mAP is 0.0, which could be the possible problem ??
你好, 我在使用多GPU训练的时候, 每次都会遇到这个问题
我使用的训练命令是
我用了4个Tesla T4 GPU, 而且4张卡都是空闲状态,为什么会出现显存不足的现象呢?