ubuntu20.04安装显卡驱动,cuda,cuDNN,pycuda,tensorrt教程

服务	版本号	下载地址
显卡驱动	460.106.00	https://www.nvidia.cn/download/driverResults.aspx/182489/cn/
cuda	11.2.2	https://developer.nvidia.com/cuda-11.2.2-download-archive
cuDNN	8.1.0	https://developer.nvidia.com/rdp/cudnn-archive
Pycuda	2022.2.2	https://pypi.org/project/pycuda/
tensorrt	8.5.3	https://developer.nvidia.com/nvidia-tensorrt-8x-download

1. 安装显卡驱动

1.1查看显卡型号

ubuntu-drivers devices

1.2禁用nouveau

vi/etc/modprobe.d/blacklist.conf

末尾添加

blacklist nouveau

options nouveau modeset=0

更新配置

update-initramfs -u

reboot重启

查看nouveau是否禁用(没有输出就是禁用了)

lsmod | grep nouveau

1.3安装

sh NVIDIA-Linux-x86_64-460.106.00.run-no-x-check -no-nouveau-check -no-opengl-files

参数说明：

1) –no-x-check：表示安装驱动时不检查X服务，非必需，我们已经禁用图形界面。

2) –no-nouveau-check：表示安装驱动时不检查nouveau，非必需，我们已经禁用驱动。

3) –no-opengl-files：表示只安装驱动文件，不安装OpenGL文件。这个参数不可省略，否则会导致登陆界面死循环，英语一般称为”login loop”或者”stuck in login”。

安装过程中的选项

（1）The distribution-provided pre-install script failed! Are you sureyou want to continue? 选择 yes 继续。

（2）Would you like to register the kernel module souces with DKMS? Thiswill allow DKMS to automatically build a new module, if you install a differentkernel later? 选择NO继续

（3）Would you like to run the nvidia-xconfigutility to automaticallyupdate your x configuration so that the NVIDIA x driver will be used when yourestart x? Any pre-existing x confile will be backed up. 选择 Yes 继续

（4）Install 32-Bit compatibility libraries? 选择 No 继续

1.4查看显卡驱动是否安装完成

nvidia-smi

2. 安装cuda(deb方式)

2.1安装deb

dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.2-460.32.03-1_amd64.deb

后面apt-get install cuda

出现了报错

执行apt --fix-broken install

完成之后安装cuda

apt-get install cuda

安装完成之后

2.2配置环境变量

vim ~/.bashrc

末尾添加两行

export PATH=/usr/local/cuda-11.2/bin${PATH:+:${PATH}}

exportLD_LIBRARY_PATH=/usr/local/cuda11.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source ~/.bashrc

更新环境变量

2.3查看cuda是否安装完成

nvcc -V

注意！！！！！！！！！！！！！！！

想要TensorRT安装成功，cuda必须是deb安装

不然安装TensorRT的时候会出现软件包依赖问题

3.安装cuDNN

提前把这三个deb包下载好

3.1安装三个deb包

注意先后顺序，不然可能会有依赖问题

3.2测试cnDNN

cd /usr/src/cudnn_samples_v8/mnistCUDNN/

执行make clean 和make install

执行./mnistCUDNN

出现Test passed 说明安装完成。

4.安装pycuda

需安装python3

apt install python3

只需一条命令即可

pip install 'pycuda>=2022.2.2'

5.安装TensorRT(deb方式)

5.1 安装dpkg包

dpkg -inv-tensorrt-local-repo-ubuntu2004-8.5.3-cuda-11.8_1.0-1_amd64.deb

5.2 生成密钥

apt-key add /var/nv-tensorrt-local-repo-ubuntu2004-8.5.3-cuda-11.8/3EFA7C6A.pub

5.3更新

apt-get update

5.4安装tensorrt

apt-get install tensorrt

完成！！！！！