My environment is Ubuntu 16.04, with tensorflow-gpu 1.13.X, cuda 10.1, cudnn 7.6. I have already configured tensorRT 126.96.36.199 well. In python, all the import package are fine, such as:
import tensorflow as tf
But when I try to save TensorRT model, there is the INFO: tensorflow : Running against TensorRT version 0.0.0. I searched the 5.X document which lists that cudnn and tf version should be 7.5 and 1.12.
Could you give me some advises to solve this problem? or should I have to degrade my tensorflow and cudnn?