TensorRT安装

Posted 洪流之源

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了TensorRT安装相关的知识,希望对你有一定的参考价值。

  下面以TensorRT-8.0.1为例,基于ubuntu20.04安装,默认已经安装GPU驱动、CUDA、cuDNN;TensorRT依赖CUDA、cuDNN的版本,比如TensorRT软件包TensorRT-8.0.1.6.Linux.x86_64-gnu.cuda-11.3.cudnn8.2.tar.gz,依赖于cuda11.3、cudnn8.2,因此操作系统中必须事先安装好对应版本的cuda、cudnn。

1. 下载TensorRT

TensorRT8.0下载链接:https://developer.nvidia.com/nvidia-tensorrt-8x-download

TensorRT的安装方式有多种,本次采用TAR包的形式安装,根据系统已经安装的CUDA版本下载对应的TAR包,根据已经安装的CUDA版本,选择相应的下载链接,因为已经安装了cuda11.3,因此选择了如下红框标注版本:

以上是关于TensorRT安装的主要内容,如果未能解决你的问题,请参考以下文章

ubuntu安装tensorrt

TensorRT安装记录(8.2.5)

TensorRT安装

TensorRT安装

通过 tensorrt 测试但无法导入 tensorrt

AI性能优化之TensorRT(1 tensorrt简介及安装)