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:
I get the error:
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:
and that works.
I am confused on whether this is an issue with the installation of tensorRT or it is tensorflow specific.