ubuntu16.04配置深度学习GPU环境
一、安装nvidia显卡驱动
1、上英伟达官网下载和自己机子匹配的显卡驱动
https://www.nvidia.com/zh-cn/geforce/
2、禁用nouveau驱动
sudo gedit /etc/modprobe.d/blacklist.conf
在末尾加上下面语句
blacklist nouveau
options nouveau modeset=0
保存退出
sudo update-initramfs -u
重启
登录命令行界面:ctrl+alt+F1
输入用户名密码
找到驱动所在路径
关掉图形界面
赋予文件执行权限,并sudo运行
sudo chmod 777 NVIDIA-Linux-x86_64-390.67.run
sudo service lightdm stop
sudo ./NVIDIA-Linux-x86_64-390.67.run
安装成功后重启,并检测
nvidia-smi
二、安装cuda9.0
上官网下载cuda9.0
运行安装
三、安装cudnn7
官网下载cudnn7
tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/