ubuntu 安装驱动及CUDA教程
一.检查
若电脑已经有驱动,必须先删除原来驱动
使用以下指令
#for case1: original driver installed by apt-get:
sudo apt-get remove --purge nvidia*
#for case2: original driver installed by runfile:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run –uninstall
若没有驱动,则无需这一步
二.查找
lspci | grep -i nvidia
查看显卡型号
sudo dpkg --list | grep nvidia-*
查看驱动版本
三.下载驱动以及CUDA
从nvidia官网下载对应版本驱动和CUDA 网址链接https://www.nvidia.com
本人下载的驱动是NVIDIA-Linux-x86_64-390.87.run(首先找到显卡型号并查看对应版本的驱动,如果版本错误后面执行驱动文件也会提示需要安装哪个版本正确),下载的CUDA是CUDA9.0,
四.禁用nouveau驱动
执行以下指令
sudo gedit /etc/modprobe.d/blacklist.conf
在文本最后添加:(禁用nouveau第三方驱动,之后也不需要改回来)
blacklist nouveau
blacklist lbm-nouveau
options nouveau modeset=0
alias nouveau off
alias lbm-nouveau off
再执行
sudo update-initramfs -u
重启电脑
执行
lsmod | grep nouveau
如果无错误,说明禁用nouveau成功
五.禁用X-Window服务
sudo service lightdm stop
这会关闭图形界面,出现以下界面
按Ctrl-Alt+F1进入命令行界面,如果直接输入指令会出现login incorect
所以先输入电脑用户名,按enter键输入密码,即可以像终端一样输入指令
六.命令行安装驱动
输入以下指令
sudo chmod +x NVIDIA-Linux-x86_64-384.59.run
(给驱动run文件赋予执行权限)
sudo ./NVIDIA-Linux-x86_64-390.87.run –no-opengl-files
(后面的参数非常重要,不可省略)
然后一直按enter键,安装驱动成功后,输入
sudo service lightdm start,恢复到图形界面,即电脑主页面
在执行以下命令验证是否安装成功
nvidia-smi (若列出GPU的信息列表,表示驱动安装成功)
七.安装CUDA
执行下面命令
sudo ./ cuda_9.0.176_384.81_linux.run—no-opengl-libs
之后按照提示依次输入
accept (同意安装)
n (不安装Driver,因为已安装最新驱动)
y (安装CUDA Toolkit)
点击 (安装到默认目录)
y (创建安装目录的软链接)
n (不复制Samples,因为在安装目录下有sample)
安装后检验是否成功,输入
cat /usr/local/cuda/version.txt(列出CUDA版本即是安装成功)
八.安装cudnn
从invidia官网下载cudnn,
下载之后,解压cudnn压缩包文件
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
执行下面指令,其实就是拷贝头文件和库文件并给予权限
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/usr/local/cuda/lib64/libcudnn*
安装完成后,执行以下指令检验是否成功
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
出现cudnn版本显示,即成功安装
全文结束,感谢借鉴。