TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard

Question: Will TensorRT 9 be available in the 23.11 NGC container ?

Open MatthieuToulemont opened this issue 2 years ago • 7 comments

Hello, thanks for all the great work !

Some of my models require bfloat16 at inference time, I saw it was added in TensorRT 9 with TensorRT-LLM, and I was wondering when TensorRT 9 will be available as the default version in the NGC containers.

Thank you,

MatthieuToulemont avatar Nov 09 '23 17:11 MatthieuToulemont

@nvpohanh ^ ^

zerollzeng avatar Nov 12 '23 12:11 zerollzeng

23.11 NGC container will still use TRT 8.6 since all the TRT 9.x releases are only recommended for LLM workloads. We will update the TRT version in the NGC container when there is a general TRT release for all use cases. Thanks

nvpohanh avatar Nov 12 '23 14:11 nvpohanh

Thank you, do you know when bfloat16 will be added to TensorRT then ?

bloat16 has become so widespread in training that it feels unfortunate not to have it available on the fastest inference framework. Even more so when a model trained in bfloat16 does not work in float16. Happened to me a couple of times :/

MatthieuToulemont avatar Nov 13 '23 09:11 MatthieuToulemont

We will update the TRT version in the NGC container when there is a general TRT release for all use cases

Do you have an estimated released date ?

MatthieuToulemont avatar Nov 13 '23 09:11 MatthieuToulemont

Having any guideline on how adapting a BF16 trained model for a TensorRT (version < 9) would be super useful in the meantime

david-PHR avatar Nov 13 '23 10:11 david-PHR

Any update on this ?

can we hope for the 24.02 NGC container ?

it's mainly related to this issue for my side: https://github.com/NVIDIA/TensorRT/issues/3124

copasseron avatar Feb 22 '24 15:02 copasseron

There won't be NGC container with TRT 9+ until Q2. Until then, please build the container using: https://github.com/NVIDIA/TensorRT/blob/v9.3.0/docker/ubuntu-22.04.Dockerfile

Thanks

nvpohanh avatar Feb 23 '24 03:02 nvpohanh

closing since no activity for more than 3 weeks, thanks all

ttyio avatar Apr 16 '24 16:04 ttyio