My Jupyter notebook and Colab kernel getting crashed while converting the Tensorflow model to Tensor RT, Can anyone Please resolve this problem

I have attached my Colab notebook and the resource folder which contains the model and images.


TensorRT Version: 5.1.5
GPU Type: Colab GPU
Nvidia Driver Version: 465.89
CUDA Version: 10.1
CUDNN Version: 8.2.1
Operating System + Version: windows 10
Python Version (if applicable): 3.7
TensorFlow Version (if applicable): 2.0
PyTorch Version (if applicable):
Baremetal or Container (if container which image + tag):

Relevant Files

colab notebook for tensor to tensorrt conversion
Colab Notebook
model and resource files
model and resource folder

Steps To Reproduce

Please include:

  • Exact steps/commands to build your repro
  • Exact steps/commands to run your repro
  • Full traceback of errors encountered

Hi @Ragu_23,

Looks like you’re using old version of TensorRT. We recommend you to please try on latest TensorRT version 8.0.1. Also please make sure you are following correct installation steps.

Thank you.