bubbliiiing / yolo3-pytorch

这是一个yolo3-pytorch的源码,可以用于训练自己的模型。
MIT License
2.01k stars 585 forks source link

fatal : Memory allocation failure #188

Open Dragon1309ycc opened 9 months ago

Dragon1309ycc commented 9 months ago

fatal : Memory allocation failure 一堆引用 RuntimeError: CUDA error: unknown error CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect. For debugging consider passing CUDA_LAUNCH_BLOCKING=1. Epoch 1/200: 0%| | 0/8275 [00:11<?, ?it/s<class 'dict'>] 这个情况该怎么解决呢大佬们

Dragon1309ycc commented 9 months ago

两个batch都调到2了还是显示内存不够啊