Ubuntu20.04下CUDA11.8、cuDNN8.6.0、TensorRT8.5.3.1的配置过程

 1、 NVIDIA驱动安装

系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改。

2、CUDA的安装

NVIDIA官网CUDA下载页面下载CUDA11.8，按步骤进行安装，注意在安装CUDA的时候取消驱动安装这个选项。

安装完后配置环境变量，sudo gedit ~/.bashrc，添加：

export PATH=$PATH:/usr/local/cuda-11.8/bin

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.8/lib64

3、cudnn的安装

NVIDIA官网cuDNN下载页面下载cuDNN8.6.0的对应版本压缩文件夹。

 解压后将对应include和lib中的文件拷贝到cuda-11.8的include和lib64文件夹中：

sudo cp -r lib/* /usr/local/cuda-11.8/lib64

sudo cp -r include/* /usr/local/cuda-11.8/include

 3、TensorRT的安装

 NVIDIA官网TensorRT下载页面下载TensorRT的对应版本压缩文件夹。

 同cuDNN类似，解压后将对应include和lib中的文件拷贝到cuda-11.8的include和lib64文件夹中：

sudo cp -r lib/* /usr/local/cuda-11.8/lib64

sudo cp -r include/* /usr/local/cuda-11.8/include

4、安装验证

TensorRT解压文件夹中找到sample目录下的sampleOnnxMNIST，在该目录中make一下，并在生成bin目录下执行：./sample_onnx_mnist。

Over。