深度学习环境配置
NVIDIA驱动安装
在安装驱动之前一定需要确认一下自己的显卡的型号,AMD就不用说了,现在不支持。Nvidia显卡也需要注意一下是否为10系(1080Ti,1080,1070…),如果为10系而且主板上没有集成显卡的时候,这个时候就需要你更换一个级别较低的显卡,10系显卡安装驱动基本是会出毛病的。
1.卸载旧的驱动(新安装的系统是不需要的)
sudo apt-get purge nvidia*
2.关闭Ubuntu自带的nouveau nvidia驱动
使用终端打开文件
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
打开后把这些复制进去
blacklist nouveau
options nouveau modeset=0
然后保存进行更新
sudo update-initramfs -u
接着重启
reboot
机器打开以后使用下边的命令查看是否为空,如果是空的那么操作成功
lsmod | grep nouveau
3.驱动安装
使用快捷键Ctrl+Alt+F1 进入图形界面,登陆自己的账号(输入自己的用户名和密码)
输入下边的命令关掉图形界面
sudo service lightdm stop
找到下载好驱动存放的目录,提高文件的权限(这里驱动安装包是NVIDIA-Linux-x86_64-375.39.run,根据你自己驱动安装包设置)
sudo chmod 777 NVIDIA-Linux-x86_64-375.39.run
下边进行安装
sudo ./NVIDIA-Linux-x86_64-375.39.run -no-x-check -no-nouveau-check -no-opengl-files
–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要!!
–no-x-check 安装驱动时不检查X服务
–no-nouveau-check 安装驱动时不检查nouveau
在安装过程中有可能需要需要序列号,按照他的提示选择产生一个直接使用就是可以的。
安装结束以后再打开图形界面
sudo service lightdm start
到此,驱动安装完成。
CUDA安装
CUDA可以去NVIDAI官网中去下载,下载 run 格式,方便安装。
下载完成以后在终端中找到下载文件所在的目录进行安装
sudo sh cuda_8.0.61_375.26_linux.run
进行安装时会有很多提示信息,如果在屏幕左下角有“more”,可以按键Q直接退出。
注意:提示中会有一个询问是否需要安装NVIDIA驱动,这个是已经安装好的,选择“no”,其余选择“yes”
安装完成后,将CUDA加入环境变量
进入环境变量文件
sudo gedit ~/.bashrc
在文档的最后添加
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
文件保存关闭以后,在终端中运行下边命令,使环境变量更新
source ~/.bashrc
cuDNN安装
cuDNN安装就比较简单直接去Nvidia官网下载中心,下载CUDA相对应的cuDNN。
下载完以后直接解压,将其复制到相应的目录下,这样cuDNN 就安装完成了。
sudo cp -a cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64/