pierluigiferrari / ssd_keras

A Keras port of Single Shot MultiBox Detector
Apache License 2.0
1.86k stars 935 forks source link

The device being used is CPU while capturing image from webcam. How do I use my GPU for processing instead? #390

Closed GabrielRiente closed 2 years ago

GabrielRiente commented 2 years ago

I'm using Tensorflow 1.13.1 and Keras 2.3.1, that's what worked and I got it working. However, I only get 5 fps. I'm running the code using PyCharm and all dependencies listed. Should I try Conda or something else? How can I optimize performance using my GPU? Is it an issue with Pycharm? It's a GTX 1060 6GB with CUDA 11.7. I've installed CUDA Development Kit and cuDNN, with all environment variables set.