简要记录这几部分安装的关键点
一.安装NVIDIA驱动
1、去英伟达官网下载驱动程序(http://www.nvidia.cn/Download/index.aspx?lang=cn)
2、把 nouveau 驱动加入黑名单
sudo nano /etc/modprobe.d/blacklist-nouveau.conf
3、更新一下
sudo update-initramfs -u
4、重启系统 reboot
****这边需要着重说一下,当出现重启黑屏的情况,可尝试把显示屏连接到HDMI口,我的黑屏问题就是这么解决的。
5、确认下nouveau是否确认被禁用,使用命令: lsmod | grep nouveau
6、关闭x-window
sudo service lightdm stop
7、Ctrl+alt+F1 输入登陆用户名和密码
8、开始安装之前下载下来的驱动,按照每一步提示安装
sudo ./NVIDIA.run
9、重新启动X-Window
sudo service lightdm start
10、Ctrl+alt+F7进入图形界面
11、查看是否安装成功
nvidia-smi
二.安装cuda
开始安装cuda9.2但驱动不能支持,于是卸载,卸载方法参考
https://blog.youkuaiyun.com/yucicheung/article/details/79382797
重新装9.0版本,安装步骤参考下面链接
https://blog.youkuaiyun.com/QLULIBIN/article/details/78714596
三.安装cuDNN
cd /usr/local
sudo tar -xzvf ~/downloads/cudnn-8.0-linux-x64-v5.1.tgz
cuDNN安装完成
进一步验证现在的驱动和cuda是否可用
nvcc -V
四.安装tesorflow-gpu 1.5版本
pip install tensorflow-gpu==1.5.0