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

NVIDIA-AI-IOT / tf_trt_models

TensorFlow models accelerated with NVIDIA TensorRT

BSD 3-Clause "New" or "Revised" License

682 stars 245 forks source link

Open zcc720 opened 5 years ago

zcc720 commented 5 years ago

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

yuvaramsingh94 commented 5 years ago

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

mustafamerttunali commented 4 years ago

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!

MayurDeshmukh10 commented 4 years ago

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