TensorRT

Description

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.

Environment

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.
https://docs.nvidia.com/deeplearning/tensorrt/install-guide/index.html

Thank you.