Windows10 安装 theano、keras、tensorflow、tensorflow-gpu

最新推荐文章于 2021-01-14 01:47:56 发布

原创最新推荐文章于 2021-01-14 01:47:56 发布 · 557 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#安装theano #安装Keras #安装TensorFlow/TensorFlow-gpu #win10

本文详细介绍了如何使用pip和conda安装TensorFlow及其GPU版本，解决安装过程中的常见问题，如GraphViz配置和pydot错误。同时，提供了安装Keras及Theano的步骤，确保深度学习环境的顺利搭建。

部署运行你感兴趣的模型镜像

安装 TensorFlow 最好的参考教程就是官方文档 ，它好就好在它是中文的。

官方文档中只有 pip 的安装示例，我尝试了 conda 安装也能成功，但是版本没有用 pip 安装的新，所以会出现某些方法调用出错的问题。

安装 TensorFlow-gpu 可以参考这个 Keras 深度学习环境配置（WIN10/GPU加速版），或许能帮助你摆脱很多的坑，因为有大神已经帮我们打包好了 tensorflow-gpu 的依赖，我们只需要执行 pip install tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl 便一劳永逸。

附上该安装包的地址：tensorflow_gpu-1.9.0-cp36-cp36m-win_amd64.whl

需要要注意的是这个安装包依赖于 cuda9.2 以及 cudnn7.1，要预先安装并配置好这两个的环境变量再安装该包。

但是我按照上面这个来装 tensorfloe-gpu 的话，用到 Keras 的地方可能会报这样的错误：

OSError: pydot failed to call GraphViz.Please install GraphViz (https://www.graphviz.org/) and ensure that its executables are in the $PATH.

甚至在此之前会先报“pydot not found”，这个 pip install pydot 就好了。但是前者出的错需要在你安装好 GraghViz 并配置好环境变量之后，修改 pydot.py 的一处源码：将原来的 self.prog = 'dot' 修改成 self.prog = 'dot.exe' ，否则依然会报错，因为这个错误就是 pydot.py 找不到 GraghViz 的系统环境变量的 dot 导致的，因为实际上是 dot.exe。