gpu not used for training?

endernewton / tf-faster-rcnn

Tensorflow Faster RCNN for Object Detection

https://arxiv.org/pdf/1702.02138.pdf

MIT License

3.65k stars 1.58k forks source link

gpu not used for training? #511

Closed kashyap92 closed 2 years ago

kashyap92 commented 2 years ago

This is a great implementaion of Faster R CNN, thanks! I have a quick question.

I am using tensorflow-gpu == 1.14 as i have cuda 11(i do not have cuda 8) and while training, the model does not use gpu at all. Could you please advise? Is the code not compatible with tensorflow-gpu>1.2

Any help is much appreciated.

Thanks,

kashyap92 commented 2 years ago

Never mind, the issue is solved.

kashyap92 commented 2 years ago

resolved.