我在学习神经网络
在做的时候我想让GPU来进行训练网络
所以如何调用GPU跑程序(keras框架)???
keras有两个后台,我个人是使用tensorflow后台。
具体的就是安装tensorflow的gpu版本,很多csdn里头的博客都有介绍
環境與安裝可以參考這篇:
https://medium.com/@WhoYoung99/2018%E6%9C%80%E6%96%B0win10%E5%AE%89%E8%A3%9Dtensorflow-gpu-keras-8b3f8652509a
記得對一下相應版本:
https://www.tensorflow.org/install/source#common_installation_problems
https://www.tensorflow.org/install/source_windows
若有問題可以查:
https://stackoverflow.com/