步骤一:NVIDIA显卡(GTX1060、GTX1080)安装(注意:不能修改GCC版本,不行可能要选 >5.0的版本)
sudo ./NVIDIA-Linux-x86_64-*****.run –no-opengl-files –no-x-check –no-nouveau-check
我是全部带了,才没有Ubuntu循环登录(我当时安装全部选以上)
链接:步骤一可以参考: (我揣测opengl导致他机子Ubuntu循环登录.)
步骤二:CUDA8.0(推荐NVIDIA官网下载.run文件)
1.注意:GCC要降低为5.0以下版本(目前CUDA不支持GCC高版本的编译)
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 100
此时可以尝试编译cuda,如果正常跳过2,有问题可参考如下:
2.下载cuda8.0********.run,解压出****.pm 文件 copy到perl文件
./cuda8.0****.run --tar -mxvf
3.cuda 运行,如果报错是缺什么文件 那么apt-get重新安装4.重点!!显卡驱动不要安装,之前已经装了否则覆盖,可能导致Ubuntu循环登录
5.可以写入 vim ./.bashrc
source ~/.bashrc
生效
步骤三:cudnn v7(NVIDIA官网下载,需要注册)
注意:安装目录与cuda的目录保持一致就可以了
链接:步骤三和四可以参考:(虽然版本不一样,但是操作一样)
步骤四:pip install tensorflow-gpu (我安装已经是1.3版本了)
测试:可以拿任意程序测试。然后查看显卡运行情况,一般运行显卡占用都是98%到100%
显卡查看每10s更新:
watch -n 10 nvidia-smi
如果有问题可以留言!谢谢