【前言】这篇文章将告诉你如何在windows中安装TensorFlow转载请注明出处
文档下载位置:http://download.youkuaiyun.com/download/huangbin6/9757370
windows中安装tensorflow【译文】
选择安装哪种版本的TensorFlow
TensorFlow两种版本介绍
- CPU版:如果你的系统不包含NVIDIA的GPU,你就需要安装这个版本。【NOTE】这种版本的TensorFlow比GPU版的更容易安装(大概5-10分钟就能安装结束),所以即时你的设备已经有NVIDIA的GPU,我们也推荐你在学习的过程中使用这种版本的TensorFlow。(注意,实际情况中,为了得到更高的运算性能,往往会使用GPU进行运算。)
- GPU版:TensorFlow程序在GPU上运行的速度明显快于在CPU中运行。因此,如果你的系统有满足如下的先决条件的NVIDIA的GPU,并且你的应用对性能的要求很高,那么你应该安装这种TensorFlow版本。(注:笔者在进行后续的TensorFlow实验中均是使用的这种版本的TensorFlow)
GPU版TensorFlow的要求
要想安装GPU版本的TensorFlow,那么如下的NVIDIA软件就必须安装在你的操作系统中。(这里的操作系统当然就是windows了,笔者的操作系统是windows10),CUDA相关软件的下载需要注册NVIDIA,并且需要填一些表单,照着填写即可。此处不做过多说明,有问题可以联系笔者,相关软件笔者也是有的,联系:694968711@qq.com
- CUDA® 工具包 8.0. 更多细节, 请查阅:http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/#axzz4Yv1uz9Dv 确保你已经按照NVIDIA官方文档的描述将有关CUDA的路劲添加到%path%环境变量中。下载连接:https://developer.nvidia.com/cuda-downloads
- cuDNN v5.1. 安装参考:https://developer.nvidia.com/cudnn. 下载:https://developer.nvidia.com/cudnn (要注册Nvidia用户,并加入CuDNN开发组,填若干问卷就可以下载了)选择下载版本时要注意和Cuda版本匹配。解压后覆盖至CUDA的安装目录下(解压后对应目录放即可)
Python安装
TensorFlow在Windows上只支持64位Python3.5,可以通过Python 3.5 from python.org 或 Python 3.5 from Anaconda 下载并安装Python3.5.2(注意选择正确的操作系统)。(这里笔者选择的是python3.5 from python.org,即从官网下载的python)python的安装比较简单,注意python的版本即可。
你必须选择安装TensorFlow的方式。有如下两种方式可供用来安装TensorFlow:
- "native" pip(此处尽介绍这种安装方式)
- Anaconda
本地pip方式安装TensorFlow是直接在操作系统上安装,并不是在一个虚拟容器中安装。因此需要操作系统具备要求的python等环境。如果你本地系统具备这样的条件,那么本地pip的方式安装TensorFlow只是一条简单的命令而已。(使用这种方式安装,任何目录下的tensorflow程序都可以运行,笔者使用的就是这种安装方式)
【注】如果不能翻墙,则只能选择离线安装,即下载tensorflow的包到本地,然后使用pip命令安装。1.GPU版本,2.CPU版本正常安装命令如下:
-
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0rc0-cp35-cp35m-win_amd64.whl
-
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-0.12.0rc0-cp35-cp35m-win_amd64.whl
官方pip安装方式:pip3 install –upgrade tensorflow pip3 install –upgrade tensorflow-gpu
安装完成后验证
打开命令提示符