Ubuntu 18.04安装nvidia驱动、cuda、cudnn、tensorrt

补充: ubuntu22.04搭建pytorch环境

1、安装 nvidia 驱动，需要重启生效

nvidia-smi #查看驱动信息
sudo apt install nvidia-XXX

2、安装CUDA
下载与pytorch版本一致的cuda，并安装:
链接: https://developer.nvidia.com/cuda-toolkit-archive

配置环境变量
sudo vim ~/.bashrc，末尾添加

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

nvcc -V 命令查看 cuda信息

3、安装cuDNN
cuDNN 下载相应版本的cudnn，需要登陆。

下载.deb可以直接安装。

下载是库则将cuda/include/cudnn.h文件复制到usr/local/cuda/include文件夹，将cuda/lib64/下所有文件复制到/usr/local/cuda/lib64文件夹中，并添加读取权限：

sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r/usr/local/cuda/lib64/libcudnn*