ubuntu16.04Nvidia驱动、CUDA、cuDNN安装与卸载

时间:2024-03-04 09:53:00

(一)Nvidia驱动安装

ubuntu 16.04默认已安装第三方驱动程序nouveau,首次安装nvidia显卡驱动首先需要禁用nouveau,不然会碰到冲突的问题,导致无法安装nvidia显卡驱动。
1、打开blacklist.conf

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

(若目录下没有blacklist.conf,则创建该文件)

2、在blacklist.conf文件最后部分插入以下两行内容

blacklist nouveau
options nouveau modeset=0

3、更新系统

sudo update-initramfs -u

4、重启系统(一定要重启)

sudo reboot

5、验证nouveau是否已禁用

lsmod | grep nouveau

没有任何输出则禁止成功

6、开始安装驱动

方案1:与cuda一起安装(推荐)下载路径:https://developer.nvidia.com/cuda-downloads

关闭图形化界面(不然会报"You appear to be running an X server"错误)

sudo service lightdm stop 

安装命令(这个run文件可以cuda和NVIDIA驱动一起安装)

sudo sh cuda_10.0.130_410.48_linux.run

安装成功后,会显示installed,否则会显示failed

方案2:当然也可以分开安装(不推荐,因为驱动版本与cuda版本有关系)

首先下载驱动文件,然后

sudo chmod 777 NVIDIA-Linux-x86_64-418.43.run

安装NVIDIA驱动(注意参数)

sudo ./NVIDIA-Linux-x86_64-375.20.run –no-opengl-files

–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要

–no-x-check 安装驱动时不检查X服务

–no-nouveau-check 安装驱动时不检查nouveau

后面两个参数可不加;

安装完成后重启即可。

7、安装完后

sudo service lightdm start

sudo reboot

(二)Nvidia驱动卸载
首先ctrl+Alt+F2/F1进入命令行状态,然后停止lightdm

sudo service lightdm stop

卸载命令位置/usr/bin/nvidia-uninstall,以下命令即可卸载。

sudo /usr/bin/nvidia-uninstall

不找这个命令的位置,也可以

sudo apt-get install remove --purge nvidia* 

卸载完后输入nvidia-smi命令没有任何反应即卸载成功。重启电脑可能会出现图形桌面字体变得异常大,这是正常情况,因为电脑没有显卡驱动,当重装驱动后就正常了。

(三)cuda安装

1、安装命令

sudo apt-get install remove --purge nvidia* 

(这条命令可以驱动和cuda一起安装)

2、设置环境变量
终端中输入:

sudo gedit /etc/profile

在打开的文件末尾,复制添加以下两行:

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

3、验证是否安装成功

验证CUDA Toolkit,输入:

nvcc -V

(四)cuda卸载

cuda的默认安装在 /usr/local/cuda-8.0下,用下面的命令卸载:

sudo /usr/local/cuda-8.0/bin/uninstall_cuda-8.0.pl 

会出现长时间的刷屏现象,说明cuda文件其实还蛮多的。

/usr/local/cuda-8.0没有自动删除,需要手动完成删除。

 下载安装文件
按需求下载cudnn的安装文件:https://developer.nvidia.com/rdp/cudnn-archive

(五)安装cudnn

cp  cudnn-10.0-linux-x64-v7.3.1.20.solitairetheme8 cudnn-10.0-linux-x64-v7.3.1.20.tgz
tar -xvf cudnn-10.0-linux-x64-v7.3.1.20.tgz

解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令:

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

(3)查看cudnn版本
在终端输入

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