search

hhk7734 / tensorflow-yolov4

YOLOv4 Implemented in Tensorflow 2.
MIT License
136 stars 75 forks source link

How to use GPU for Inferencing? #79

Closed JayantGoel001 closed 3 years ago

JayantGoel001 commented 3 years ago

Hey @hhk7734 Can we GPU's for inferencing the yolov4 model?

hhk7734 commented 3 years ago

I tested on Titan RTX and Tesla V100(colab).

JayantGoel001 commented 3 years ago

I tested on Titan RTX and Tesla V100(colab).

So I don't need to do any extra to use GPU?

hhk7734 commented 3 years ago

Whether or not to use a GPU has little to do with this library. Most of them are TensorFlow settings, and in general, TF will automatically assign the model to the GPU.

hhk7734 commented 3 years ago

https://github.com/hhk7734/tensorflow-yolov4/blob/3d31292cefba198b1528a90b3f435204efe7be73/py_src/yolov4/tf/__init__.py#L45-L48

Maybe, only this part is related to GPU settings.

JayantGoel001 commented 3 years ago

Ok Yup Got It Thanks 😁