目前从tensorrt官网上只能找到ubuntu18.04的版本,但是我的操作系统是ubuntu20.04,实际上直接尝试直接安装ubuntu18.04的版本,这边测试是没有问题
官网下载地址:https://developer.nvidia.com/nvidia-tensorrt-download
我下载的是TensorRT 7这个版本,根据自己需要选择cuda、tensorrt对应的版本,我下载的是这个版本
https://developer.nvidia.com/compute/machine-learning/tensorrt/secure/7.2.3/tars/TensorRT-7.2.3.4.Ubuntu-18.04.x86_64-gnu.cuda-11.1.cudnn8.1.tar.gz
注意我们下载的时候下载tar包的版本,下载完成后解压该压缩包
我本地是python3.8,所以直接安装
pip install tensorrt-7.2.3.4-cp38-none-linux_x86_64.whl
然后进行测试
python3进入python交互环境
import tensorrt as trt trt.__version__ "7.2.3.4"
打印上述版本信息说明tensorrt安装成功了。