(一)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