本文将解释如何在Windows下安装TensorFlow。
需要先确定哪种类型的TensorFlow:
如果在系统中安装使用GPU运行的TensorFlow,需要确保下面介绍的NVIDIA软件已经安装到系统中。
%PATH%
环境变量中。%PATH%
环境变量中。如果系统中已经安装了以前的相关包,请更新到所指定的版本。
在安装TensorFlow之前必须选定一个安装机制。目前提供2种机制:
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,输入以下命令:
C:\> pip3 install --upgrade tensorflow
安装GPU版本的TensorFlow,使用以下命令:
C:\> pip3 install --upgrade tensorflow-gpu
Anaconda模式安装
再次强调,Anaconda安装是有第三方社区提供的,非官方。
在Anaconda环境中安装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" |