写到最开头:
开始下载和配置前,一定要先看一下我后面描述的几个坑!坑!!坑!!!
关于Keras
支持卷积神经网络和循环神经网络,或者两者融合。
支持CPU和GPU.
Python要求 2.7-3.6.
我这里安装的是TensorFlow 2.0和最新的 TensorFlow GPU2.1
这两个库的安装方法参照:写的非常详细
python+tensorflow+tensorflow-gpu+CUDA+cuDNN+pycharm全套环境
我这里说几个其中没有提到的坑
坑1
其中对于CUDA的版本安装,要先在安全模式下卸载原有的NVIDIA显卡驱动程序。先安装CUDA,之后再安装对于的最新版显卡驱动程序。
清理程序我这有:
在操作系统安全模式下运行ddu,选择显卡品牌,点击清理即可。
然后在重启回到正常模式,安装CUDA。安完再装新版显卡驱动。
坑2
TensorFlow版本,CUDA版本,cuDNN版本的匹配。
一定要先看好自己的匹配,再去下载!!!!不要见到最新版本就下,几个最新版本之间也不是相互匹配的。
干货:TensorFlow1.2~2.1各个GPU版本CUDA和cuDNN对应版本整理
坑3
最后就是,电脑有GPU再装TensorFlow GPU版本。没有也不用费劲了,装CPU版本就好了。
如何看有没有GPU和验证TensorFlow GPU版本是否ok,可以跑一下我上一节,MNIST数据集使用的代码。看一下任务管理器,性能里面的GPU使用情况就行了。运行时会有起伏。就OK了
未运行代码:
跑训练时:
log中也能看到:
类似这句有显示开启了显卡就OK了。