This commit is contained in:
Rémi Bédard-Couture 2024-05-18 16:05:15 -04:00
commit 95ebe0c8be

View File

@ -6,6 +6,7 @@ import numpy as np
try:
import tensorrt as trt
from cuda import cuda
TRT_VERSION = int(trt.__version__[0 : trt.__version__.find(".")])
TRT_SUPPORT = True