原文出处： http://xugaoxiang.com/post/117

软硬件环境

• ubuntu 16.04 64bit
• NVidia GTX 1070
• CUDA 9.0

安装驱动

从官网下载RUN文件,首先屏蔽开源nouvear驱动,编辑/etc/modprobe.d/blacklist-nouveau.conf文件,加入内容

blacklist vga16fb
blacklist nouveau
blacklist rivafb
blacklist nvidiafb
blacklist rivatv

如果你原来安装过驱动,最好先删除

sudo apt-get --purge remove nvidia-*
sudo apt-get --purge remove xserver-xorg-video-nouveau

NVIDIA的驱动无法在GUI的环境下安装,所以得切换到命令行,按组合快捷键Ctrl + Alt + F1

sudo /etc/init.d/lightdm stop
sudo ./NVIDIA-Linux-x86_64-390.25.RUN

安装过程基本上就是选中yes or no,装完之后执行

sudo update-initramfs -u

安装CUDA

同样的,我们也进入命令行模式安装cuda

sudo /etc/init.d/lightdm stop
sudo ./cuda_9.0.176_384.81_linux.run
sudo /etc/init.d/lightdm start

然后配置下,编辑~/.bashrc

最后

source ~/.bashrc

安装cudnn

到官网https://developer.nvidia.com/cudnn去下载,这里需要注册,有点麻烦

tar xvf cudnn-9.1-linux-x64-v7.1.tgz
cd cuda
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cd -a cuda/lib64/libcudnn*  /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*

测试

安装完毕,我们来测试下环境是否都ok

刚好我这里跑了示例,所以你能看到gpu基本上是满负荷工作

参考资料

http://www.nvidia.com/Download/index.aspx?lang=en-us
https://developer.nvidia.com/cuda-toolkit
https://blog.csdn.net/hanging_gardens/article/details/64439344