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

TensorRT is not enabled! HELP ME!!!

Open EEECGWood opened this issue 5 years ago • 3 comments

When I use tensorRT to optimize my tensorflow graph , (trt_graph = trt.create_inference_graph( ... input_graph_def=frozen_graph, ... outputs=output_names, ... max_batch_size=1, ... max_workspace_size_bytes=1 << 25, ... precision_mode='FP16', ... minimum_segment_size=50 ... )) I got errors like this: INFO:tensorflow:Running against TensorRT version 0.0.0 Traceback (most recent call last): File "", line 7, in File "/root/anaconda3/lib/python3.6/site-packages/tensorflow/contrib/tensorrt/python/trt_convert.py", line 153, in create_inference_graph int(msg[0])) tensorflow.python.framework.errors_impl.FailedPreconditionError: TensorRT is not enabled!

I have found many methods to solve this problem, however, no one seems useful, can you help me ? I really need to solve this problem as soon as possible !

EEECGWood avatar Mar 27 '19 03:03 EEECGWood

Which versión of tensorflow, cuda and tensort do you have installed?

isra60 avatar Mar 27 '19 06:03 isra60

Which versión of tensorflow, cuda and tensort do you have installed?

I use tensorflow 1.10.0 cuda 9 tensorrt 5.0.2.6 cudnn 7.3.1

EEECGWood avatar Mar 28 '19 01:03 EEECGWood

Which is your hardware? Jetson Tx2?? If tha is the case you have a very updated version of TensorRT. If you are on Jetpack 3.3 you should stay in TensorRT4

isra60 avatar Mar 28 '19 07:03 isra60