Unable to export hdf5 to etlt after Tao Training on Colab

Accelerated Computing Intelligent Video Analytics TAO Toolkit
9

I cannot run exact steps as yours since suddenly the colab told me that it cannot connect to gpu backend due to usage limits.
But I find a way for you to debug or run something directly inside the docker.
You can run
!tao model yolo_v4 run /bin/bash

image
image1640×348 35.5 KB

then, import the tensorrt. You can find as below screentshot, the trt can be imported.

image
image1511×348 28.7 KB

You can try to use this way to run export.

/content# yolo_v4 export xxx

BTW, I download TensorRT-8.6.1.6.Linux.x86_64-gnu.cuda-12.0.tar.gz from the official download center. So, you can see the trt version is 8.6.1.