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

作者: 大细菌 | 来源:发表于2017-08-30 19:32 被阅读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)//图像处理相关

    相关文章

      网友评论

        本文标题:Windows环境下搭建TensorFlow(GPU)简单教程

        本文链接:https://www.haomeiwen.com/subject/rkkadxtx.html