I wanted to convert my tensorflow model to tensorrt but while I ran a conversion script it shoots an issue regarding unidentified Cuda device.
TensorRT Version: 6.0
GPU Type: 1 GPU. Tesla t4
CUDA Version: 10.0
CUDNN Version: 7.6.5
Operating System + Version: Ubuntu 18.04
Python Version (if applicable): 3.6
TensorFlow Version (if applicable): 2.1 (GPU supported)
Resource used to convert to TensorRT :
Steps To Reproduce
() Any Tensorflow model
() Run conversion script which can be found on the link mentioned above.
Can't identify the cuda device. Running on device 0 Segmentation fault (core dumped)
Thanks in advance.