Open ckolluru opened 4 days ago
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?
10.2.0.post1
TRT_MAIN_VERSION
TRT_MAIN_VERSION = int(trt.__version__.split('.')[0])
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])