ubuntu20.04安装tensorrt

目前从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包的版本,下载完成后解压该压缩包

图片.png

我本地是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安装成功了。

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://sulao.cn/post/900.html