I am trying to convert mine TensorFlow model to TF-TRT but I am getting an error.
ModuleNotFoundError: No module named ‘tensorflow.python.compiler.tensorrt’
Line in code: ‘from tensorflow.python.compiler.tensorrt import trt_convert as trt’
OS: Windows 10
GPU: GTX 1070
TRT Version: 184.108.40.206
I have already used this machine to train models on GPU and it is working fine so CUDA is installed correctly. TRT is installed by the instructions from Nvidia documentations.