深度学习 从零开始 —— 神经网络(三),安装配置keras,GPU版本TensorFlow,安装CUDA各种坑~~~

写到最开头:

开始下载和配置前,一定要先看一下我后面描述的几个坑!坑!!坑!!!

关于Keras

官网https://keras.io/

支持卷积神经网络和循环神经网络,或者两者融合。
支持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了。

猜你喜欢

转载自blog.csdn.net/wwb1990/article/details/104870255
今日推荐