ImportError: cannot import name 'tensorrt' from 'tensorflow.python.compiler'

AI & Data Science Deep Learning (Training & Inference) TensorRT
1

Hi,
I have a virtual environment where I have installed latest tensoflow-gpu (1.14.0) using pip command. I have followed the process as per documentation for for installing
Cuda 10.1
cDNN 7.5
TensorRT 5.1.5
Tensorflow GPU 1.14

I am using Windows 10. I am trying to use TensorRT.

from tensorflow.python.compiler.tensorrt import trt_convert as trt

but I am getting following error

ImportError: cannot import name ‘tensorrt’ from ‘tensorflow.python.compiler’ (C:\Users\AIKAKI\Anaconda3\envs\openCV\lib\site-packages\tensorflow\python\compiler_init_.py)

I also checked the site-package and don’t any file named tensorRT etc.

When I checked a mirror repo, I see the file

https://git.codingcafe.org/Mirrors/tensorflow/tensorflow/tree/bdb5c9a16f84cfe3909568ae91a0315937909811/tensorflow/python/compiler/tensorrt

How do I run my code? Please help

2

TensoRT in TensorFlow is inside ‘contrib’. You can import this as follows:

from tensorflow.contrib.tensorrt import trt_convert