keras遇到的bug记录

起初是在简书上写一些笔记，但无奈简书文艺青年比较多，今天开始在CSDN记录下自己的一些学习过程。

好吧，第一篇竟然是记录bug。。。。

 

之前一直是用tf和keras，后来想学习一下caffe，所以换了cuda和cudnn，重新装了tensorflow和keras，但以前用的好好的今天确遇到了一些bug。

 

tensorflow遇到的问题是目前TF还不支持CUDA9.1，索性我又换到了CUDA9.0，但github上有人提供了支持9.1的whl文件，传送门如下：

https://github.com/fo40225/tensorflow-windows-wheel

————————————————————————————————————————————

keras遇到的问题是，在测试mnist的时候，mnist数据集下载的问题 。

from keras.datasets import mnist
(X_train, y_train), (X_test, y_test) = mnist.load_data()

如果用keras自带的函数读取数据集，不但需要，而且还会出现一堆进度条，就像下面这样：

                                      

就是这样，根本不明白这输出的是什么鬼。。。。Stack Overflow上也有人提问了，但貌似没解决方案，想看的去看下吧，传送门如下：https://stackoverflow.com/questions/51159965/connectionreseterror-winerror-10054-an-existing-connection-was-forcibly-close

各种渠道都查了，问了，但依然查到原因和解决办法！！扎心！！

干脆就换了TF的读取mnist方式，用keras搭建模型和训练，一切正常了。

from tensorflow.examples.tutorials.mnist import input_data  
mnist = input_data.read_data_sets("E:/database/mnist", one_hot=True)  

就这样了，反正以后尽量不去用keras了，因为从业人员一般不会用keras，一般都是用tensorflow、caffe（C++）、pytorch、darknet等。

后续会把简书上写的几篇搬到CSDN上来。