deep_sort_yolov3 icon indicating copy to clipboard operation
deep_sort_yolov3 copied to clipboard

Published 20 hours ago • verified publisher icon Qidian213

How to use GPU with this?

Open sumopal23 opened this issue 6 years ago • 2 comments

Hi, I'm running this code on my CPU machine. How do I make it run on the GPU ? Maybe I'm missing something but couldn't find a documentation on how to do it. I have CUDA and CUDNN installed. Thanks

sumopal23 avatar Oct 02 '18 16:10 sumopal23

maybe your tensorflow is not gpu version. My computer use the tensorflow-gpu 1.9.0 is all right. you can have a try.

404hasbeenfound avatar Oct 08 '18 06:10 404hasbeenfound

use conda to instal tensorflow-gpu. Much easier as they automatically install CUDA and CUDNN.

abinjoabraham avatar Mar 15 '19 13:03 abinjoabraham