Prepare for TensorRT 8.6
The upcoming CUDA-12 upgrades requires TensorRT 8.6 and this version has a new set of headers which requires an update to the bazel configure script. I also had to change `find_cuda_config.py` because previously it has only been reporting the major version of TensorRT back to the configure script. But we will also need the minor version to distinguish between TensorRT 8.5 and below, and TensorRT 8.6+. PiperOrigin-RevId: 564927510
Showing
想要评论请 注册 或 登录