Tensorflow 安装
学习资料:
- 官方关于 Tensorflow 安装的 说明
安装 Tensorflow 时需要注意的几点:
- MacOS, Linux, Windows 系统均已支持 Tensorflow
- 确定你的 python 版本
- 你的 GPU 是 NVIDIA, 就可以安装 GPU 版本的 Tensorflow; 你的 GPU 不是 NVIDIA 也没有关系, 安装 CPU 版本的就好了.
多种安装途径 ¶
Tensorflow 的安装方式很多. 比如官网提供的:
这节内容使用 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
然后重复这个安装教程的步骤, 从头安装新版本.
降低知识传递的门槛
莫烦很常从互联网上学习知识,开源分享的人是我学习的榜样。 他们的行为也改变了我对教育的态度: 降低知识传递的门槛。 免费 奉献我的所学正是受这种态度的影响。 通过 【赞助莫烦】 能让我感到认同,我也更有理由坚持下去。
想当算法工程师拿高薪?转行AI无门道?莫烦也想祝你一臂之力,市面上机构繁杂, 经过莫烦的筛选,七月在线脱颖而出, 莫烦和他们合作,独家提供大额 【培训优惠券】, 让你更有机会接触丰富的教学资源、培训辅导体验, 祝你找/换工作/学习顺利~