【必看】TensorFlow 2 官方安装指南

这篇《TensorFlow 2 安装指南》节选自由谷歌开发者专家（GDE）李锡涵@snowkylin 编写、广受好评的开源 TensorFlow 入门教程——《简单粗暴TensorFlow 2》，详细介绍了在个人电脑或服务器上安装 TensorFlow 2 的步骤和各种细节，让你第一次安装 TensorFlow 2 就上手！

 

文 /  李锡涵，Google Developers Expert

本文节选自《简单粗暴 TensorFlow 2.0》

 

一般安装步骤

TensorFlow 的 Python 版本使用最为广泛。作为一个 Python 包，TensorFlow 和其他 Python 包的安装方法区别不大，使用 Python 的包管理器 pip 即可。具体步骤如下：

    1. 安装 Python 环境。此处建议安装 Anaconda 。这是一个开源的 Python 发行版本，提供了一个完整的科学计算环境，包括 NumPy、SciPy 等常用科学计算库。当然，你有权选择自己喜欢的 Python 环境。

    2. 使用 Anaconda 自带的 conda 包管理器建立一个 Conda 虚拟环境，并进入该虚拟环境。在命令行下输入以下命令：

# 建立虚拟环境

# “tf2.0”是你建立的 Conda 虚拟环境的名字

# python=3.7 代表该虚拟环境的 Python 版本为 3.7

conda create --name tf2.0 python=3.7

# 进入名为“tensorflow”的虚拟环境

conda activate tf2.0     

 

    3. 使用 Python 包管理器 pip 安装 TensorFlow。在命令行下输入以下命令：

pip install tensorflow==2.0.0-beta1     # TensorFlow CPU 版本

或

pip install tensorflow-gpu    # TensorFlow GPU 版本，需要具有 NVIDIA 显卡及正确安装驱动程序，详见下文

等待片刻即安装完毕。

 

 

小技巧

也可以使用 conda install tensorflow 或者 conda install tensorflow-gpu 来安装 TensorFlow，不过 conda 源的版本往往更新较慢，难以第一时间获得最新的 TensorFlow 版本；

在 Windows 下，需要打开开始菜单中的 “Anaconda Prompt” 进入 Anaconda 的命令行环境；

在国内环境下，推荐使用 国内的 pypi 镜像 和 Anaconda 镜像 ，将显著提升 pip 和 conda 的下载速度；

如果对磁盘空间要求严格（比如服务器环境），可以安装 Miniconda ，仅包含 Python 和 Conda，其他的包可自己按需安装。

 

pip 和 conda 包管理器

pip 是最为广泛使用的 Python 包管理器，可以帮助我们获得最新的 Python 包并进行管理。常用命令如下：

pip install [package-name]              # 安装名为 [package-name] 的包

pip install [package-name]==X.X         # 安装名为 [package-name] 的包并指定版本 X.X

pip install [package-name] --proxy=代理服务器IP:端口号         # 使用代理服务器安装

pip install [package-name] --upgrade    # 更新名为 [package-name] 的包

pip uninstall [package-name]            # 删除名为 [package-name] 的包

pip list