search

umyelab / LabGym

Quantify user-defined behaviors.
GNU General Public License v3.0
56 stars 4 forks source link

Could not find TensorRT #164

Open odong8322OU opened 4 days ago

odong8322OU commented 4 days ago

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!

yujiahu415 commented 4 days ago

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?