pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
https://pytorch.org/TensorRT
BSD 3-Clause "New" or "Revised" License
2.6k stars 351 forks source link

Add support for JetPack 6.1 build #3211

Closed lanluo-nvidia closed 1 month ago

lanluo-nvidia commented 1 month ago

Description

Add support for the jetpack6.1 build. Currently jetpack 6.1 has: cuda: 12.6 python: 3.10 tensorrt: 10.3 DLFW: 24.08 (pytorch: 2.5.0)

It can build successfully on release/2.5 branch, will fail on the main branch for now, as main branch has used the pytorch method from 2.6.0 which is not in the DLFW24.08 distribution.

Type of change

Please delete options that are not relevant and/or add your own.

Checklist:

lanluo-nvidia commented 1 month ago

TODO: align with which pytorch to target (ngc or the stable ones)

lanluo-nvidia commented 1 month ago

Need to add the new tutorial to the index.rst for it to show up in the docs

added.