search

NVIDIA-AI-IOT / tf_to_trt_image_classification

Image classification with NVIDIA TensorRT from TensorFlow models.
BSD 3-Clause "New" or "Revised" License
455 stars 155 forks source link

Error while running python scripts/convert_plan.py #24

Open gangooteli opened 6 years ago

gangooteli commented 6 years ago

root@61dd12ecedc9:/home/tf_to_trt_image_classification# python scripts/convert_plan.py data/frozen_graphs/mobilenet_v1_0p25_128.pb data/plans/mobilenet_v1_0p25_128.plan input 128 128 MobilenetV1/Logits/SpatialSqueeze 1 0 float Using output node MobilenetV1/Logits/SpatialSqueeze Converting to UFF graph No. nodes: 306 UFF Output written to data/tmp.uff cudnnLayerUtils.cpp (288) - Cuda Error in smVersion: 35 terminate called after throwing an instance of 'nvinfer1::CudaError' what(): std::exception

I am getting error while calling convert_plan.py

Please suggest. Using below versions: Cuda 9.0 Ubuntu 16.04 Tensorrt 3