pycharm环境调用tensorrt

### 配置和使用 TensorRT 的步骤 #### 安装必要的依赖项 为了能够在 PyCharm 中顺利配置并使用 TensorRT，安装合适的 Python 版本及其对应的库至关重要。如果尚未安装 Anaconda 或 Miniconda，可以通过清华镜像站加速下载过程[^2]。 ```bash wget -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py39_4.10.3-Linux-x86_64.sh ``` 创建一个新的 conda 虚拟环境来隔离项目所需的软件包： ```bash conda create --name tensorrt_env python=3.9 conda activate tensorrt_env ``` 接着，在此环境中安装 TensorRT 和其他必需的 Python 库，比如 NumPy 等基础科学计算工具。 #### 设置 PYTHONPATH 变量 为了让 Python 解释器能够找到 TensorRT 提供的功能模块，需调整 `PYTHONPATH` 环境变量以便加入本地机器上的特定路径[^3]。这一步骤确保了即使是在虚拟环境下也能访问到系统级安装好的 TensorRT 库。 在 Linux/MacOS 上执行如下命令设置临时环境变量；对于 Windows 用户，则应采用相应的 PowerShell 命令或通过图形界面完成相同操作。 ```bash export PYTHONPATH=$PYTHONPATH:/usr/local/lib/python3.x/site-packages/ ``` 请注意替换 `/usr/local/lib/python3.x/site-packages/` 为实际存放 TensorRT Python 接口的位置。 #### 编写代码以调用 TensorRT 有了上述准备工作之后，现在可以在 PyCharm 内编写一段简单的脚本来加载 ONNX 文件，并将其转换成 TensorRT 引擎格式。这里给出一个基于 YOLOv5 模型的例子[^5]。 ```python import tensorrt as trt from polygraphy.backend.trt import CreateConfig, EngineFromNetwork, TrtRunner from polygraphy.comparator import Comparator def build_engine(onnx_file_path): TRT_LOGGER = trt.Logger(trt.Logger.WARNING) with trt.Builder(TRT_LOGGER) as builder, \ builder.create_network(1 << int(trt.NetworkDefinitionCreationFlag.EXPLICIT_BATCH)) as network, \ trt.OnnxParser(network, TRT_LOGGER) as parser: config = builder.create_builder_config() config.max_workspace_size = 1 << 30 with open(onnx_file_path, 'rb') as model: if not parser.parse(model.read()): for error in range(parser.num_errors): print(parser.get_error(error)) return None engine = builder.build_serialized_network(network, config) return engine if __name__ == "__main__": onnx_model_path = "models/yolov5s.onnx" # 构建 TensorRT 引擎 serialized_engine = build_engine(onnx_model_path) # 使用 Polygraphy 运行推理 runner = TrtRunner(EngineFromNetwork(serialized_engine)) results = Comparator.run([runner]) ``` 这段程序展示了如何读取 `.onnx` 文件并通过 TensorRT 创建优化后的引擎对象。最后利用第三方库 Polygraphy 来简化推理流程[^4]。