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

本文详细介绍了Keras及TensorFlow的安装过程,包括在不同版本的CUDA和cuDNN下正确匹配TensorFlow版本的方法,以及在有GPU环境下安装TensorFlowGPU版本的注意事项。
摘要由CSDN通过智能技术生成

写到最开头:

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

关于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了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值