Linux上TensorRT环境配置【以YOLOX为例】

原创

已于 2023-06-07 22:16:41 修改 · 2.3k 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#linux #ubuntu #python

于 2023-06-06 14:50:48 首次发布

本文记录了在Ubuntu20.04上配置YOLOX的TensorRT环境的过程，包括安装ONNX、ONNXRuntime、ONNX-Simplifier，以及解决CMake找不到可执行文件的问题。此外，还介绍了安装apex的注意事项和TensorRT的详细安装步骤，包括添加库到环境变量。在安装torch2trt时遇到的库文件缺失问题也得到了解决，通过将库文件复制到指定目录来修复。

前言

以YOLOX的tensorrt环境配置为例，记录安装步骤以及过程中遇到的一些问题

Linux OS：Ubuntu 20.04
CUDA Toolkit（cuDNN）：11.6
python：3.9.16
pytorch/torchvision

问题一：安装onnx

onnx==1.8.1
onnxruntime==1.8.0
onnx-simplifier==0.3.5

安装使用清华镜像源：

-i https://pypi.tuna.tsinghua.edu.cn/simple

按照onnx代码库的readme.txt执行

apt-get install python3-pip python3-dev libprotobuf-dev protobuf-compiler

export CMAKE_ARGS="-DONNX_USE_PROTOBUF_SHARED_LIBS=ON"

使用pip安装上面库时遇到AssertionError: Could not find "cmake" executable!
按照博文：【pip按照onnx报错：AssertionError】的说法，cmake需要安装在/usr/bin/这个目录下，经过检查，未安装Cmake！！！

安装Cmake
Linux下安装CMake，【2023-6-6】最新的版本为3.26.4，安装的方式一共有三种：