Hardware configuration for running YoloV5 model

Hi,

We are using YoloV5x model for object detection, we would like to know the hardware configuration for running the model in a local machine.

Thanks,
Sreekanth

@ptrblck

Do you have any idea on the hardware configuration required for running the Yolov5x model.

No, I don’t have any configurations ready, but the repository mentions the training was performed on a V100 16GB, so you might use this device as a reference.

Hi @ptrblck

Thanks. We are not planning to train the model in local machine. When we run the model in local machine (OS: Ubuntu) the prediction time is more with dedicated GPU.

Wanted to know which GPU will suit to run the model in local machine.