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

前言

以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，安装的方式一共有三种：

• 通过软件包仓库安装
  Ubuntu安装命令为：

  sudo apt install cmake
  

  虽然能成功安装，但是版本较旧。
  卸载命令：

  sudo