tf_trt_models icon indicating copy to clipboard operation
tf_trt_models copied to clipboard

Published 20 hours ago • verified publisher icon NVIDIA-AI-IOT

ModuleNotFoundError: No module named 'tensorflow.contrib.tensorrt'

Open zcc720 opened this issue 6 years ago • 3 comments

hi, I got an error when running this code line: import tensorflow.contrib.tensorrt as trt can you tell me how to slove this?

zcc720 avatar Nov 30 '18 07:11 zcc720

please use latest version of tensorflow gpu . by default Tf gpu are coming with trt . i am currently using tf 1.8 and it is working fine

yuvaramsingh94 avatar Mar 05 '19 09:03 yuvaramsingh94

please use latest version of tensorflow gpu . by default Tf gpu are coming with trt . i am currently using tf 1.8 and it is working fine

Tried latest version of Tensorflow(2.0). Doesn't work!

mustafamerttunali avatar Oct 02 '19 15:10 mustafamerttunali

For Tensorflow(2.0) use - from tensorflow.python.compiler.tensorrt import trt_convert as trt This worked for me.

MayurDeshmukh10 avatar Jun 23 '20 13:06 MayurDeshmukh10