操作系统:ubuntu16.04
GPU:GTX1080
一、安装显卡驱动
1. 首先查看GPU版本信息
lspci | grep -i nvidia
2. 验证自己的Linux版本是否支持 CUDA(Ubuntu 16.04和Ubuntu17.04都支持)
uname -m && cat /etc/*release
3. 验证是否安装gcc
gcc --version
如果没有则输入 $ sudo apt-get install gcc下载
sudo apt-get install gcc
4. 验证系统是否安装了kernel header和 package development
uname -r
sudo apt-get install linux-headers-$(uname -r)
5. 卸载已安装驱动
sudo apt remove --purge nvidia*
6.禁用nouveau驱动
在终端运行
lsmod | grep nouveau
如果有输出,说明nouveau正在运行,需要手动禁用
在/etc/modprobe.d中创建文件blacklist-nouveau.conf
输入
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
在文件中输入一下内容:
blacklist nouveau
options nouveau modeset=0
执行
sudo update-initramfs -u
查看nouveau是否已经成功禁用
lsmod | grep nouveau
然后$ sudo reboot重启
重启后进入用户登录界面时(不要登录进桌面),按Ctrl+Alt+F1进入字符终端界面,输入用户名和密码进行登录,登录成功后,再次查看。
关闭图形界面
sudo service lightdm stop
7.安装驱动
这里采用apt安装驱动
添加Graphic Drivers PPA
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get -update
安装nvidia driver,我安装的驱动版本是387,之前装的nvidia-384在mxnet跑程序时出现错误,提示版本太低。
输入
sudo apt-get install nvidia-387
安装完成后重启
sudo reboot
重启后可以执行
sudo nvidia-smi
查看驱动版本和显卡型号
二、安装cuda
在英伟达官网上下载cuda9.1, https://developer.nvidia.com/cuda-downloads
按图中的选择下载即可
在/home/downloads里面可以看到下载的cuda文件
执行
sudo sh cuda9.1.85_387.26_linux.run
之后就是很长的一段说明,一直按回车即可。
当提示“是否为NVIDIA安装驱动?”时选择“N”,不然会把之前的驱动覆盖掉。其他选择默认即可。
设置环境变量
sudo vim ~/.bashrc
来编辑.bashrc配置文件
在文件最后加上下面几句话(直接复制就行)
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
然后设置环境变量和动态链接库,在命令行输入:
sudo vim/etc/profile
在打开的文件末尾加入:
export PATH = /usr/lacal/cuda/bin:$PATH
保存之后,创建链接文件
sudo gedit /etc/ld.so.conf.d/cuda.conf
在打开的文件中添加如下语句:
/usr/local/cuda/lib64
然后执行
sudo ldconfig
重启一下机子
三、安装cudnn
首先在官网上下载 https://developer.nvidia.com/rdp/form/cudnn-download-survey
进去要注册登录
下载第一个for CUDA9.1的
首先解压tgz文件
sudo -xzvf cudnn-9.1-linux-x64-v7.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 /usr/local/cuda/lib64/libcudnn*
即可完成cudnn安装
如果提示库缺失错误,可参考以下命令:
sudo cp /usr/local/cuda-9.1/lib64/libcudart.so.9.1 /usr/local/lib/libcudart.so.9.1 && sudo ldconfig
sudo cp /usr/local/cuda-9.1/lib64/libcublas.so.9.1 /usr/local/lib/libcublas.so.9.1 && sudo ldconfig
sudo cp /usr/local/cuda-9.1/lib64/libcurand.so.9.1 /usr/local/lib/libcurabd.so.9.1 && sudo ldconfig
sudo cp /usr/local/cuda-9.1/lib64/libcudnn.so.7 /usr/local/lib/libcudnn.so.7 && sudo ldconfig
新版的cudnn下载变成了deb文件了
所以直接在终端输入:
sudo dpkg -i libcudnn7_7.3.1.20-1+cuda9.0_amd64.deb
即可完成对cudnn的安装,非常简单。