I have downloaded tensorflow-gpu with pip version 9 and have downloaded tensorRT following exact nvidia instructions.
Now I am trying to run the code for resnet 50 found here:
https://devblogs.nvidia.com/tensorrt-integration-speeds-tensorflow-inference/
I get the error:
import tensorflow.contrib.tensorrt
Traceback (most recent call last):
File “”, line 1, in
ImportError: No module named tensorrt
I am running python 2 and the cuda version that I have is 9.
I ran the command in python:
import tensorrt
and that works.
I am confused on whether this is an issue with the installation of tensorRT or it is tensorflow specific.