I am using tensorRT 3.0.4 and tensorflow 1.7 to optimize the VGG model using the following code. The GPU i am using is Titan X.
trt_graph = trt.create_inference_graph(
input_graph_def=frozen_graph,
outputs=['input', 'vgg_16/fc8/squeezed'],
max_batch_size=1,
max_workspace_size_bytes=1 << 25,
precision_mode='FP16',
minimum_segment_size=50
)
After running this code I am getting the following error.
2018-07-20 10:09:24.822613: I tensorflow/core/grappler/devices.cc:51] Number of eligible GPUs (core count >= 8): 1
*** stack smashing detected ***: python terminated
Aborted (core dumped)