前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Windows安装TensorFlow 原

Windows安装TensorFlow 原

作者头像
随风溜达的向日葵
发布2018-08-15 11:37:32
6420
发布2018-08-15 11:37:32
举报
文章被收录于专栏:技术墨客技术墨客

本文将解释如何在Windows下安装TensorFlow。

确定安装哪类TensorFlow

需要先确定哪种类型的TensorFlow:

  • 仅支持CUP运算版本:如果电脑的系统没有 NVIDIA®的GPU,那么必须安装这个版本。这个版本的TensorFlow安装非常简单(安装仅需一个命令,5到10分钟),所以即使系统中有满足要求的NVIDIA® GPU官方还是建议在学习阶段安装这个版本。
  • 支持GPU运算的版本:TensorFlow程序在GPU下运行比在CPU下运行明显快很多。如果系统中包含 NVIDIA®的GPU满足下一个小节所示的条件并且程序对性能要求很高,建议安装此版本。

运行TensorFlow所需要的GPU配置

如果在系统中安装使用GPU运行的TensorFlow,需要确保下面介绍的NVIDIA软件已经安装到系统中。

  • CUDA® Toolkit 8.0。请看 NVIDIA安装cuda 的文档,根据文档中的描述确保已经将CUDA相关的路径增加到 %PATH% 环境变量中。
  • NVIDIA的驱动关联 CUDA Toolkit 8.0。
  • cuDNN v5.1。请查看 NVIDIA cudnn 文档。需要注意的是cuDNN通常安装在与其他CUDA动态链接库(dll)不同的位置。确保已经将cuDNN的 动态链接库(dll)的地址添加到系统的  %PATH% 环境变量中。
  • GPU显卡必须拥有3.0以上版本的CUDA计算能力,查看 NVIDIA显卡支持列表 了解支持情况。

如果系统中已经安装了以前的相关包,请更新到所指定的版本。

如何安装TensorFlow

在安装TensorFlow之前必须选定一个安装机制。目前提供2种机制:

  • "native"app
  • Anaconda

Native的安装(以下简称本地安装)方式会将TensorFlow直接安装在当前的系统中,不会在系统和TensorFlow之间搭建任何的虚拟环境,所以本地安装不会额外安装一个独立的容器。需要注意的是本地安装可能会干扰系统中其他基于python安装的程序。如果事先已经安装配置了满足需要的python环境,本地安装通常只需要一个命令就可以完成。使用本地安装,用户可以在系统中任何位置运行TensorFlow。

在Anaconda模式下,需要使用conda创建一个虚拟环境。官方优先推荐使用 pip install 命令来安装TensorFlow,其次再考虑anaconda的 conda install 命令。conda包是第三方社区提供的(非TensorFlow官方),TensorFlow团队从始至终都不会去测试在conda中运行的情况,在使用时需考虑这个风险。

本地安装

首先,需要安装以下版本的python:

TensorFlow在windows操作系统中仅仅支持3.5.x版本的python。Python 3.5.x附带pip3软件包管理器,这是用于安装TensorFlow的程序。

安装TensorFlow需要启动一个终端(terminal),然后在该终端中输入对应的pip3 install命令。安装仅支持CPU版本的TensorFlow,输入以下命令:

代码语言:javascript
复制
C:\> pip3 install --upgrade tensorflow

安装GPU版本的TensorFlow,使用以下命令:

代码语言:javascript
复制
C:\> pip3 install --upgrade tensorflow-gpu

Anaconda模式安装

再次强调,Anaconda安装是有第三方社区提供的,非官方。

在Anaconda环境中安装TensorFlow分为以下几个步骤:

  1. 按照 Anaconda download site 的说明进行下载和安装操作。
  2. 调用以下命令来创建一个名为tensorflow的conda环境: C:> conda create -n tensorflow
  3.  键入以下命令来启用conda环境: C:> activate tensorflow (tensorflow)C:> # Your prompt should change
  4. 键入以下命令在conda环境中安装TensorFlow。这里 安装CPU版本的命令: (tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-1.0.1-cp35-cp35m-win_amd64.whl 这是GPU版本的命令: (tensorflow)C:> pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.0.1-cp35-cp35m-win_amd64.whl

验证安装 

  1. 通过以下步骤来验证TensorFlow是否安装成功:
  2. 启动一个终端(比如CMD)
  3. 如果通过Anaconda安装,先启动Anaconda环境。
  4. 在终端运行python
  5. C:> python
  6. 在python的交互环境中输入以下脚本代码:
  7. >>> import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session() print(sess.run(hello)) 如果python输出以下内容,则表明TensorFlow已经安装成功然后就可以写TensorFlow的程序了:
  8. Hello, TensorFlow! 如果收到了一些异常信息,请继续向下看。

常见的安装问题

TensorFlow通过Stack Overflow网站来记录错误信息以及处理方法。下面的列表包含一些跳转的到 Stack Overflow的连接。如果在安装过程中遇到的问题没有在下面中,请到Stack Overflow去搜索相关的关键字。若还是搜索不到,请直接提出新问题并标记 tensorflow 的标签。

Stack Overflow Link

Error Message

41007279

[...\stream_executor\dso_loader.cc] Couldn't open CUDA library nvcuda.dll

41007279

[...\stream_executor\cuda\cuda_dnn.cc] Unable to load cuDNN DSO

42006320

ImportError: Traceback (most recent call last): File "...\tensorflow\core\framework\graph_pb2.py", line 6, in from google.protobuf import descriptor as _descriptor ImportError: cannot import name 'descriptor'

42011070

No module named "pywrap_tensorflow"

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2017/04/18 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 确定安装哪类TensorFlow
  • 运行TensorFlow所需要的GPU配置
  • 如何安装TensorFlow
    • 本地安装
      • 验证安装 
        • 常见的安装问题
        相关产品与服务
        容器服务
        腾讯云容器服务(Tencent Kubernetes Engine, TKE)基于原生 kubernetes 提供以容器为核心的、高度可扩展的高性能容器管理服务,覆盖 Serverless、边缘计算、分布式云等多种业务部署场景,业内首创单个集群兼容多种计算节点的容器资源管理模式。同时产品作为云原生 Finops 领先布道者,主导开源项目Crane,全面助力客户实现资源优化、成本控制。
        领券
        问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档