过程参考了http://blog.csdn.net/continueoo/article/details/71404066。
安装NVIDIA驱动
禁用自带的nouveau NVIDIA驱动
使用如下命令创建文件
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
文件中添加如下内容:
blacklist nouveau
options nouveau modeset=0
使用如下命令更新:
sudo update-initramfs -u
重启系统。使用如下命令确认mouveau已经禁用。
lsmod | grep nouveau
输出内容为空。说明成功。
禁用lightdm桌面管理器
sudo /etc/init.d/lightdm stop
使用Ctrl+ Alt + F1
进入命令行界面。
安装NVIDIA驱动
切换到下载的NVIDIA驱动的目录下。运行如下命令
sudo ./Your-Nvidia-Driver.run –no-opengl-files
后面这个参数是用来解决重复登录的问题的。
启用lightdm桌面管理器
sudo /etc/init.d/lightdm start
安装CUDA® Toolkit 8.0
下载
从网站https://developer.nvidia.com/cuda-downloads下载合适版本的cuda toolkit。
安装
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo apt update
sudo apt install cuda
修改环境变量
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
验证安装成功
cat /proc/driver/nvidia/version
出现如下内容
nvcc -V
出现如下内容
其实没有验证完。
安装cuDNN v6
下载
从网站https://developer.nvidia.com/rdp/cudnn-download下载下面要安装的包
安装runtime library
sudo dpkg -i libcudnn6_6.0.21-1+cuda8.0_amd64.deb
安装developer library
sudo dpkg -i libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb
安装code samples和cuDNN Library User Guide
sudo dpkg -i libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb
验证
cp -r /usr/src/cudnn_samples_v7/ $HOME
cd $HOME/cudnn_samples_v7/mnistCUDNN
make clean && make
./mnistCUDNN
出现test passed
信息。
安装libcupti-dev library
sudo apt-get install libcupti-dev
安装gpu版本的tensorflow
sudo pip install tensorflow-gpu
结束。