packyan / PyTorch-YOLOv3-kitti

use yolov3 pytorch to train kitti

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

RuntimeError: CUDA out of memory

amitmane1 opened this issue · comments

I'm getting RuntimeError for GPU even after I have set batch size to 1 instead of mentioned in repo 16.
Could you please guide me into this for any possible correction?
GPU is having capacity of 8GB

Thank you!

commented

if you use windows, set --n_cpu to 0