Ubuntu 驱动安装（显卡驱动 CUDA CUDNN TensorRT）

1. 显卡驱动

2. CUDA

2.1 安装

链接：CUDA Toolkit Archive | NVIDIA Developerhttps://developer.nvidia.com/cuda-toolkit-archive

选择需要下载的版本，和符合自己系统的CUDA下载runfile到本地

然后sh安装即可

记得不要选安装driver

2.2  环境变量配置（版本切换）

在.bashrc文件中粘贴以下语句：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda

更新环境变量:

source ~/.bashrc

版本切换：

（1）删除已经存在的CUDA软链接

sudo rm -rf cuda

sudo rm /usr/local/cuda

查看链接：

ls -l /usr/local/cuda

（2）重新建立软链接到制定版本的CUDA

sudo ln -s /usr/local/cuda-11.3 /usr/local/cuda

（3）查询当前CUDA版本

nvcc -V

2.3 卸载CUDA 

cd  /usr/local/cuda-11.8/bin

sudo ./cuda-uninstaller

3. CUDNN

3.1 安装

链接：

https://developer.nvidia.com/rdp/cudnn-archivehttps://developer.nvidia.com/rdp/cudnn-archive（1）下载后解压

（2）然后在解压压缩包在的文件夹下（名字和复制去的cuda版本都要修改）

sudo cp cudnn-linux-x86_64-8.9.7.29_cuda11-archive/lib/* /usr/local/cuda-11.3/lib64/

sudo cp cudnn-linux-x86_64-8.9.7.29_cuda11-archive/include/* /usr/local/cuda-11.3/include/

（3）查询cudnn版本

cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

4. TensorRT

4.1 安装

链接

Log in | NVIDIA Developerhttps://developer.nvidia.com/tensorrt/download/10x