keras-YOLOv3-mobilenet icon indicating copy to clipboard operation
keras-YOLOv3-mobilenet copied to clipboard

Published 20 hours ago • verified publisher icon Adamdad

How to train the model with GPU?

Open LiuShaw opened this issue 5 years ago • 0 comments

I have use the "os.environ["CUDA_VISIBLE_DEVICES"] = "0", but the traing still use CPU. How to setting to use GPU?

LiuShaw avatar Apr 03 '19 08:04 LiuShaw