Ubuntu 安装cuda、cuDNN、tensorrt

一、安装cuda

说明： 强烈建议深度学习玩家使用官方方式安装驱动，不然后面可能会出现各种环境问题。

1. 从链接中选择合适版本下载.

https://developer.nvidia.com/cuda-toolkit-archive

2. 根据对应版本的官方提示安装.

3. 查看是否安装成功

sudo dpkg -l |grep cuda

二、安装cuDNN

1. 从链接中选择合适版本下载.

https://developer.nvidia.com/rdp/cudnn-archive

2. 安装

sudo dpkg -i cudnn-local-repo-XXX.deb
# 根据安装提示复制粘贴即可
sudo cp /var/cudnn-local-repo-XXX/cudnn-local-XXX-keyring.gpg /usr/share/keyrings/

3. 查看是否安装成功

sudo dpkg -l |grep cudnn

三、安装tensorrt

1. 从链接中选择合适版本下载.

https://developer.nvidia.com/nvidia-tensorrt-download

此处EA和GA的含义如下：

1. EA 版本代表抢先体验（在正式发布之前）。
2. GA 代表通用性。 表示稳定版，经过全面测试。

2. 安装

sudo dpkg -i nv-tensorrt-local-repo-XXX.deb
# 根据安装提示复制粘贴即可
sudo cp /var/nv-tensorrt-local-repo-XXX/nv-tensorrt-local-XXX-keyring.gpg /usr/share/keyrings/

3. 查看是否安装成功

sudo dpkg -l |grep tensorrt

四、其他

1. 禁用自带的nouveau驱动

sudo gedit /etc/modprobe.d/blacklist.conf

最下方添加以下内容

blacklist nouveau
options nouveau modeset=0

sudo update-initramfs -u
reboot

重启后运行，没有任何输出说明禁用成功

lsmod | grep nouveau

2. 删除旧的nvidia驱动

sudo apt-get purge nvidia*

3. 更换apt清华源

a. 从链接中选择对应系统版本配置

https://mirror.tuna.tsinghua.edu.cn/help/ubuntu/

b. 将/etc/apt/sources.list 文件内容替换为上面的内容
c. 执行 sudo apt update