Error while running python scripts/convert_plan.py

[gangooteli]

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