一些更新
前两天Google推出了windows版的tensorflow,现在大家可以在windows上玩tensorflow了。我用tensorflow的原因是它资源很丰富、背景很雄厚。用keras的原因是简单爽快可扩展,用python的原因是这个语言对tensor(高维矩阵)进行操作非常直观方便。
简单介绍一下windows版tensorflow的安装过程:
- 首先你会需要python以及numpy scipy blas lapack mkl 等一系列library,可以说对于不熟悉linux和python世界的同学来说这就是个噩梦,对于熟悉的来说是半个噩梦,所以我最后还是安装了Anaconda3(Python3集成环境)大礼包,从官网下载的python3没有Anaconda方便。注意,最新的TF在windows上只支持python3.
- 然后从Pypi( XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX )上下载tensorflow的最新whl包。下载前请注意,分CPU和GPU两个版本。
- 如果你下载的是GPU版,请下载安装CUDA和cuDNN。如果不安装,GPU版tf启动时会提示找不到几个dll。
- 运行 pip install <下载的whl包文件名>
- ……然后在python里面 import tensorflow as tf 应该就可以了。
如果需要用Keras框架,不要用pip install,检测到系统为windows之后会傻傻地跑去下载theano(因为以前tensorflow是没有windows版本的)。我直接从github拖源码,然后Python XXXXXXXX就可以了。到装theano那一步我Ctrl-C了。完美的方案应该是把系统检测代码删掉,让用户自己选择是要tensorflow还是theano。
import keras的时候也会遇到问题,keras检测到windows系统(同时检测不到~/.keras/keras.json文件的话)就自动切到theano后端,由于没装theano也会报错。解决方法是手动改keras源码,我已经向作者提交了pull request(更新:作者没有通过request,有需要自己动手吧)。
CUDA的驱动程序在我这边有点问题,GT640M显卡在tf.Session()的一瞬间会跳出一个错误,提示我的显卡无法弹出,然后把我整个系统卡死。所以我暂时还是在用CPU训练。