服务 | 版本号 | 下载地址 |
显卡驱动 | 460.106.00 | https://www.nvidia.cn/download/driverResults.aspx/182489/cn/ |
cuda | 11.2.2 | |
cuDNN | 8.1.0 | |
Pycuda | 2022.2.2 | |
tensorrt | 8.5.3 |
安装显卡驱动
1.1查看显卡型号
ubuntu-drivers devices
![](https://i-blog.csdnimg.cn/blog_migrate/1b7afb27fdb959744f2ae330a2845ecd.png)
1.2禁用nouveau
vi/etc/modprobe.d/blacklist.conf
末尾添加
blacklist nouveau
options nouveau modeset=0
![](https://i-blog.csdnimg.cn/blog_migrate/11aaaaec43100ab6c7612eb29a57a013.png)
更新配置
update-initramfs -u
![](https://i-blog.csdnimg.cn/blog_migrate/8cd084738a53037e471007c4c27305f0.png)
reboot重启
查看nouveau是否禁用(没有输出就是禁用了)
lsmod | grep nouveau
![](https://i-blog.csdnimg.cn/blog_migrate/0b763208844aa39f713e19f5584b0f52.png)
1.3安装
sh NVIDIA-Linux-x86_64-460.106.00.run-no-x-check -no-nouveau-check -no-opengl-files
![](https://i-blog.csdnimg.cn/blog_migrate/ec04328a454ac9845d726c5ce38473ad.png)
参数说明:
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
![](https://i-blog.csdnimg.cn/blog_migrate/5d0a10525b054e9449c66f4c9a4ee419.png)
安装cuda(deb方式)
2.1安装deb
dpkg -i cuda-repo-ubuntu2004-11-2-local_11.2.2-460.32.03-1_amd64.deb
![](https://i-blog.csdnimg.cn/blog_migrate/156125aacc8a720d0d1a006dfa2860c6.png)
后面apt-get install cuda
![](https://i-blog.csdnimg.cn/blog_migrate/89f610d9307a32ec604b9113c6d243d8.png)
出现了报错
执行apt --fix-broken install
![](https://i-blog.csdnimg.cn/blog_migrate/06fe6b0d62eeadaf12f5ef3857a8e73a.png)
完成之后安装cuda
apt-get install cuda
![](https://i-blog.csdnimg.cn/blog_migrate/5d8d1498b00b38850336dd7839047d1e.png)
安装完成之后
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}}
![](https://i-blog.csdnimg.cn/blog_migrate/c1f60dcc6fa9e25f16b26e2b9a3cc57a.png)
source ~/.bashrc
更新环境变量
2.3查看cuda是否安装完成
nvcc -V
![](https://i-blog.csdnimg.cn/blog_migrate/57f08cbcbc3cf4b76be69fdd22d10734.png)
注意!!!!!!!!!!!!!!!
想要TensorRT安装成功,cuda必须是deb安装
不然安装TensorRT的时候会出现软件包依赖问题
3.安装cuDNN
![](https://i-blog.csdnimg.cn/blog_migrate/0f000d79f2ad2c6e93f0e50c97eb78de.png)
提前把这三个deb包下载好
3.1安装三个deb包
![](https://i-blog.csdnimg.cn/blog_migrate/1cc03e47313ecbc2bf0c26469b134676.png)
注意先后顺序,不然可能会有依赖问题
3.2测试cnDNN
cd /usr/src/cudnn_samples_v8/mnistCUDNN/
![](https://i-blog.csdnimg.cn/blog_migrate/b894ec594cb4fa36571c3716d9648151.png)
执行make clean 和make install
![](https://i-blog.csdnimg.cn/blog_migrate/8c4e77e0741a8e11377cb5d93a8960a8.png)
执行./mnistCUDNN
![](https://i-blog.csdnimg.cn/blog_migrate/9ae048486332f93f03008fd72371b3ef.png)
出现Test passed 说明安装完成。
4.安装pycuda
需安装python3
apt install python3
![](https://i-blog.csdnimg.cn/blog_migrate/634a342c54d76884a72b709075bc007b.png)
只需一条命令即可
pip install 'pycuda>=2022.2.2'
![](https://i-blog.csdnimg.cn/blog_migrate/4c0929914f0cbd19377b97869ef3a422.png)
5.安装TensorRT(deb方式)
5.1 安装dpkg包
dpkg -inv-tensorrt-local-repo-ubuntu2004-8.5.3-cuda-11.8_1.0-1_amd64.deb
![](https://i-blog.csdnimg.cn/blog_migrate/c4f1818be93205cba2c220079af154dd.png)
5.2 生成密钥
apt-key add /var/nv-tensorrt-local-repo-ubuntu2004-8.5.3-cuda-11.8/3EFA7C6A.pub
![](https://i-blog.csdnimg.cn/blog_migrate/d10ea004ad7090428e497a894e1faed7.png)
5.3更新
apt-get update
![](https://i-blog.csdnimg.cn/blog_migrate/5c3755829497cda125c9d584c33374fd.png)
5.4安装tensorrt
apt-get install tensorrt
![](https://i-blog.csdnimg.cn/blog_migrate/853f5c345ee7fe81eb8098cfe8fa462d.png)
完成!!!!!