原文出处： 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
