Tensorflow 安装

学习资料:

• 官方关于 Tensorflow 安装的 说明

安装 Tensorflow 时需要注意的几点:

1. MacOS, Linux, Windows 系统均已支持 Tensorflow
2. 确定你的 python 版本
3. 你的 GPU 是 NVIDIA, 就可以安装 GPU 版本的 Tensorflow; 你的 GPU 不是 NVIDIA 也没有关系, 安装 CPU 版本的就好了.

多种安装途径 ¶

Tensorflow 的安装方式很多. 比如官网提供的:

• Pip 安装
• Docker 安装

这节内容使用 pip 在每个系统的安装方式:

Linux 和 MacOS ¶

本文将提到第一种最简单的安装方式, pip 安装. 使用 pip 安装的时候要确保你的 pip 已经存在于你的电脑中. 如果还没有安装 pip. 你可以在 Terminal 窗口中运行这个, 升级必要的组件:

# Ubuntu/Linux 64-位 系统的执行代码:
$ sudo apt-get install python-pip python-dev

# Mac OS X 系统的执行代码:
$ sudo easy_install --upgrade pip
$ sudo easy_install --upgrade six

CPU 版

激动人心的时刻到了, Tensorflow (0.12之后) 做了更新, 绕过了复杂的安装步骤, 如果你只需要安装 CPU 版本的 Tensorflow, 运行下面这个就好了:

# python 2+ 的用户:
$ pip install tensorflow

# python 3+ 的用户:
$ pip3 install tensorflow

注意: 你需要8.1或更高版的 pip 才能顺利安装.

GPU 版

Tensorflow 已经不再支持 mac 的 GPU 版了, 下面是 Linux 安装 GPU 版的说明. 说先安装 NVIDIA CUDA 必要组建.

$ sudo apt-get install libcupti-dev

然后确保你的 linux 上 pip 是可用的, 接着我们可以直接通过pip 安装:

$ sudo apt-get install python-pip python-dev   # for Python 2.7
$ sudo apt-get install python3-pip python3-dev # for Python 3.n

然后选择你想要cpu 或者 gpu 版本.

$ pip install tensorflow      # Python 2.7; CPU support (no GPU support)
$ pip3 install tensorflow     # Python 3.n; CPU support (no GPU support)
$ pip install tensorflow-gpu  # Python 2.7;  GPU support
$ pip3 install tensorflow-gpu # Python 3.n; GPU support

Windows ¶

tf 在 windows 的官方安装说明 , 其实就简单的方法就是装个 Anaconda, 省了你安装这安装那, 不知道还有什么没安装的烦恼. 如果你喜欢用 pip 安装, 通过西面这个途径就好了.

安装前的检查:

• 目前只支持 Python 3.5/3.6 (64bit) 版本
• 你有安装 numpy (没有的话,请看这里numpy 安装教程)

接下来惊心动魄啦! 在 command 窗口中执行

# CPU 版的
C:\> pip3 install --upgrade tensorflow

# GPU 版的
C:\> pip3 install --upgrade tensorflow-gpu

注意

• Windows 运行 Tensorflow 如果遇到这个报错:

Error importing tensorflow.  Unless you are using bazel,
you should not try to import tensorflow from its source directory;
please exit the tensorflow source tree, and relaunch your python interpreter
from there.

不要惊慌, 尝试下载安装 Windows 的 Microsoft Visual C++ 2015 redistributable update 3 64 bit. 就能解决这个问题.

• 或者在 Windows 运行的时候出现了如下报错, 你需要安装 Windows 的 Visual C++ Redistributable for Visual Studio 2015 就能成功解决问题.

ImportError: No module named '_pywrap_tensorflow_internal'

测试 ¶

然后打开你的 python 编辑器, 输入

import tensorflow

运行脚本来检查一下是否有正确安装.

更新 Tensorflow ¶

最后, 如果你需要升级 Tensorflow 的版本, 推荐的方式是:

根据你的 python 版本, 在 terminal 中删除原有的版本

# 如果你是 Python 2, 请复制下面
pip uninstall tensorflow

# 如果你是 Python 3, 请复制下面
pip3 uninstall tensorflow

然后重复这个安装教程的步骤, 从头安装新版本.