检查 TensorRT 的版本

TensorRT 是 NVIDIA 推出的用于高性能深度学习推理的库，它能够显著提升推理速度。在开发中，我们需要检查 TensorRT 的版本，以保证代码在不同的环境下运行的正确性。以下是几种检查 TensorRT 版本的方式：

1. 在 Python 中使用 TensorRT 包中的版本号

在 Python 中使用 TensorRT 包时，可以通过以下代码获取 TensorRT 版本号：

import tensorrt as trt
print("TensorRT version: {}".format(trt.__version__))

2. 在 C++ 代码中使用 API 获取版本号

在 C++ 代码中使用 TensorRT API 时，可以通过以下代码获取 TensorRT 版本号：

#include <iostream>
#include <NvInfer.h>

int main() {
    std::cout << "TensorRT version: " << NV_TENSORRT_VERSION_MAJOR << "."
              << NV_TENSORRT_VERSION_MINOR << "."
              << NV_TENSORRT_VERSION_PATCH << std::endl;
    return 0;
}

3. 使用命令行查看版本号

在命令行中使用 TensorRT 时，可以通过以下命令获取 TensorRT 版本号：

$ tensorrt --version

以上是几种常见的检查 TensorRT 版本的方式。使用这些方法可以帮助我们在开发时及时发现问题，并确保代码的正确性。

参考链接：

TensorRT documentation

GitHub issue