Bug in TensorRT version

[ckolluru]

I was looking into your repo for creating a TensorRT engine and found this line.

https://github.com/asmirnou/watsor/blob/master/watsor/engine.py/#L14

TRT_MAIN_VERSION = int(trt.__version__[0])

If a newer version of TensorRT (say 10.2.0.post1) is used, TRT_MAIN_VERSION will be set to 1.
To set it correctly, maybe set it up this way?

TRT_MAIN_VERSION = int(trt.__version__.split('.')[0])