人工智能人工智能/模式识别/机器学习精华专题认知科学探索与创新

Windows环境下搭建TensorFlow(GPU)简单教程

2017-08-30  本文已影响570人  大细菌

一、环境

我也清楚具体原因,Windows下搭建TensorFlow特别矫情。经踩坑实验,将我自己搭建成功的步骤记录下来,具有一定的泛用性

- Windows 10 64位操作系统

- VC++ 2015(VC++ 2015 运行库)

- CUDA8.0(NVIDIA显卡)

- cuDNN

- Anaconda(Python 3.5)


二、VC++ 2015

由于我本身就有安装VS2015,故推断安装「VC++ 2015 运行库」即可


三、CUDA8.0与cuDNN

其实没有严格的安装顺序,没有NVIDIA显卡的话不能安装CUDA,但TensorFlow仍能用电脑的CPU来计算。

安装CUDA8.0,下载地址:https://developer.nvidia.com/cuda-downloads

cuDNN是NVIDIA的深度学习SDK,下载地址:https://developer.nvidia.com/cudnn

一定要下载图中圈出的版本,不然会死的很惨

将下载好的cuDNN.zip解压,将其中的bin、include、lib中的文件拷贝到CUDA安装目录下相应的文件夹,CUDA的安装路径C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

其他可运行版本供参考,来自https://github.com/tensorflow/tensorflow/issues/9469

- tensorflow 1.2 + cudnn 5.1: works fine

- tensorflow 1.2 + cudnn 6.0: ImportError

- tensorflow 1.3 + cudnn 5.1: works fine

- tensorflow 1.3 + cudnn 6.0: works fine


四、Anaconda(Python 3.5)

据说Python3.5比较稳妥,我没测试

不要按官方版了,不止是配环境变量那么简单。Anaconda集成了很多实用的库,环境变量直接配好了,建议使用

相应版本 Anaconda3-4.2.0-Windows-x86_64

安装好以后,按「win」+「R」输入cmd,呼出控制台,在小黑窗里输入“python”,如果能够正常显示python版本说明安装成功


五、安装TensorFlow

安装Anaconda后我们有了更好的控制台,「Anaconda prompt」

不了解的情况下,不要随便手贱升级pip

通过pip安装GPU版的命令

pip3 install --upgrade tensorflow-gpu

安装完毕后,在Python下输入命令,没报错就是安装成功

import tensorflow as tf

也可以采用官方的例子进行验证

Anaconda安装完python3.5后,在安装tensorflow时出现 “Cannot remove entries from nonexistent file c:\program files\anaconda3\lib\site-packages\easy-install.pth” 的问题。查看原因是因为setuptools版本太低,tensorflow要求29.0.1,当前版本为27.2.0,在更新setuptools版本的时候又找不到easy-install.pth,导致更新失败

运行:pip install --upgrade --ignore-installed setuptools,问题解决!

此处参考http://blog.csdn.net/fool_frog/article/details/53422460


六、可能需要的组件

image (1.5.5)//图像处理相关

上一篇下一篇

猜你喜欢

热点阅读