augmentedstartups / yolov3workflow

This workflow aims to make it easy to train and execute Yolo v3 object detection on Windows.
116 stars 102 forks source link

GPU is not being used during detection, showing CUDA not available #7

Open spaul13 opened 5 years ago

spaul13 commented 5 years ago

I have installed the CUDA 10.1 and corresponding cuDNN 7.5.0 and added to the path variables and everything else according to the documentation

_But while running the detection, I check the output of torch.cuda.isavailable() which is returning false and also checked the GPU utilization is 0-1% so GPU is not being used.

Can anyone please tell me how to make GPU to be used during detection to make detection faster?