【大坑啊，已出坑】ubuntu tensorflow/keras 使用gpu

【大坑啊，已出坑】ubuntu tensorflow/keras 使用gpu

一望一抹泪～ 一试一滴血～
这是我自学机器学习以来第二大坑，不是一个坑，是“坑群”。

———————–静心，静心，别着急————————

当你看到这里时，不管你做到了哪一步，请先做如下尝试，不成再详细阅读后面的东西：
1、试试这个

sudo apt-get install libcupti-dev

2、重启试试（两种情况需要重启：一是，驱动安装后再终端输入nvidia-smi，告诉你有驱动但是用不了；二是，安装NVIDIA；三是，很有可能还有别的关于驱动的情况）（别忘了先再浏览器收藏本网页）
3、试试这个小东西安装NVIDIA

———————–静心，静心，别着急————————

• 我的初始情况：我用ubuntu17.04，装的nvidia驱动、cuda、cudnn、tensorflow、keras，用系统自带的python2.7可以使用keras的fit，但是巨慢，一查发现用的是cpu而非gpu。
• 我的目标：让tensorflow/keras使用gpu运算。
• 最终结果：成功了。cpu需要40分钟的运算，在用gpu之后只需要280秒。
• 先说tips：【！！！都不要用最新的版本！！！无论ubuntu、cuda、cudnn】我在还用Ubuntu17.04时遇到无数光怪陆离的bug。由于cudnn不支持ubuntu17.04，我怀疑就是这个原因导致不成功的。所以重装了系统，在Ubuntu下用brasero很方便刻录系统盘。sudo apt install brasero
• 准备材料：
  
  • cuda：cuda 8.0（安装方式就在下载按钮下面挨着，网页留着，先不装。）
  • cudnn：cudnn v5

• 步骤：

  • 安装驱动：安装NVIDIA
  • gcc降版本（我也不知道具体为什么，别的攻略上说太高了不支持）sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++
  • 安装cuda：安装方式就在下载按钮下面挨着，之后cat /usr/local/cuda/version.txt如果出现版本号，就证明安装成功。我的是

  sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
  sudo apt-get update
  sudo apt-get install cuda
  或者，在线安装也很快：
  

  sudo dpkg -i cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda

  • 安装cudnn，并配置环境变量：这个链接只看步骤6和7，别的有错误啊！，之后cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2我也不知道出现什么算证明安装成功，但如果显示什么什么不存在则肯定没装好。如果不行，不妨重启试试！
  • 上面按照链接中做你会发现nvidia-settings和nvidia-smi和文中说的效果不一样！！不是你错了！！是文章忘了让你重启了！！亲测有效！！（坑啊坑啊坑啊啊啊啊啊）
  • 安装tensorflow：sudo pip install tensorflow-gpu==1.2不要忘了：
    
    • 1、后面的“-gpu”，我就是在装了tensorflow而非tensorflow-gpu的情况下尝试用gpu的，废了好大的力气，肯定不行啊！！（坑啊坑啊坑啊啊啊啊啊）
    • 2、后面的“==1.2”，如果没有装的最新版本，就不支持啊！！（坑啊坑啊坑啊啊啊啊啊）
  • 安装keras等常用模块
  • 进入ipython，如果import tensorflow出现import error，请参考这个链接不知道对不对你的症，我的心血啊，百度了多少东西我才发现的这个啊，最后是在nvidia官网的官方安装教程发现的
• some other tips：
  
  • 重启试试
  • 虽然装的是tensorflow-gpu，但是在程序中还是import tensorflow
• 反正我到这里是行了，如果你不行，就给我留言。

祝你好运！少踩坑！