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安装的主要内容,如果未能解决你的问题,请参考以下文章