Win10下 TensorFlow 安装配置
之前,TensorFlow还不支持Window系统,虽然可以安装,但是极其麻烦。Google在2016年11月在开发者博客中宣布新版本TensorFlow0.12,增加了对Window系统的支持。
安装前准备:
TensorFlow有两个版本CPU和GPU版本。GPU版本需要安装CUDA和cuDNN的支持,CPU版本直接安装即可。首先查看电脑是否支持CUDA。确保你的Python版本是3.5 64位。确保你有稳定的网络连接。确保你的pip版本 >= 8.1。用 pip -V 查看当前 pip 版本,用 python -m pip install -U pip 升级pip 。
然后可以开始下载 CUDA 和 cuDNN 的安装包了,注意版本号分别是 CUDA 8.0 和 cuDNN 5.1,这是 Google 官方推荐的,这里不多做介绍。
安装配置:
本人的机器支持GPU,所以我安装的是GPU版本。
安装CUDA8.0
将下载的文件解压,运行安装程序,安装过程屏幕可能会闪烁,安装时间有点长。安装之后系统变量会自动为你添加上。
测试一下是否安装成功,命令行输入nvcc –V ,看到版本信息就表示安装成功了。
安装cuDNN5.1
cuDNN是压缩包,解压之后放在需要存放的位置,然后添加环境变量到path里。
配置环境变量path:
按照官方的安装步骤实际上这里已经完成了,但是经过我的安装发现,这样如果你 import tensorflow 的话它还是找不到cuDNN的文件,也就是说下面紫色框中那一行是失败的:
这里写图片描述
然后发现 GitHub 上果然有人提交了一个一样的 issue,大家可以在这里看下,然后是这么解决这个问题的:将下面这些文件复制到相应位置,
除此之外,还有一个办法就是将C:\cuda\bin 也加进Path 环境变量里,经过测试这样也是可行的。
安装TensorFlow
通过pip命令安装,保证网络连接正常:
Pip install --upgrade --ignore-installed https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl
安装/版本/pip/cuD/NN/支持/TensorFlow/GPU/环境/变量/
安装/版本/pip/cuD/NN/支持/TensorFlow/GPU/环境/变量/
-->