Keras的model怎么使用GPU keras调用gpu

转载

killads 2024-05-14 10:33:37

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

当你看到这里时，不管你做到了哪一步，请先做如下尝试，不成再详细阅读后面的东西：
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
准备材料：

安装驱动：安装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

或者，在线安装也很快：

Keras的model怎么使用GPU keras调用gpu_python

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不要忘了：