GPU bug using tensorflow 2.x Object Detection API


I’m following this installation guide for object detection using tensorflow 2, my goal is to train a CNN using my GPU. Tensorflow seems to recognize it after the GPU support section and everything runs smoothly. However, after I install the Object Detection API, tensorflow just starts ignoring it and runs on CPU. Any help would be deeply apreciated, thanks!

submitted by /u/smcsb
[visit reddit] [comments]

Leave a Reply

Your email address will not be published.