umyelab/LabGym

Could not find TensorRT

Closed this issue · 1 comments

Hi,

I am having this warning pop up whenever I load up LabGym (see screenshot below).
Screenshot 2024-07-03 at 11 28 36 AM

It still initializes the user interface, but when I try to train the detector, it takes a long time and eventually exits with 'Killed'.
Screenshot 2024-07-03 at 11 43 41 AM

Do you know why this error occurs? Thanks!

Which version of tensorflow did you use?
The program being killed is possibly because the GPU ran out of memory. Which NVIDIA GPU did you use and what is the VRAM?