ubuntu20.04安装tensorrt

目前从tensorrt官网上只能找到ubuntu18.04的版本，但是我的操作系统是ubuntu20.04，实际上直接尝试直接安装ubuntu18.04的版本，这边测试是没有问题

我下载的是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包的版本，下载完成后解压该压缩包

图片.png

我本地是python3.8，所以直接安装

pip install tensorrt-7.2.3.4-cp38-none-linux_x86_64.whl

然后进行测试

python3进入python交互环境

01.
import tensorrt as trt
02.
trt.__version__
03.
"7.2.3.4"

打印上述版本信息说明tensorrt安装成功了。

标签