Linux/Ubuntu下安装配置Tensorflow教程

摘要

之前一直用的caffe，但是最近突然有新需求，caffe对某些方面支持的不是很人性化，所以只能试试Tensorflow，毕竟Google出品必属精品，本文主要记录Ubuntu下安装Tensorflow的方法，如果是Windows或者mac环境请参考官方文档。

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

$ sudo apt-get install python-pip python-dev

GPU 版

我电脑上之前已经装过Cuda和cudnn，之前没有装的同学请参考这里安装Cuda和cudnn的部分。

然后确保你的 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 版本.

# Ubuntu/Linux 64-bit, GPU enabled, Python 2.7
# Requires CUDA toolkit 8.0 and CuDNN v5. For other versions, see "Install from sources" below.

$ export TF_BINARY_URL=https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.11.0-cp27-none-linux_x86_64.whl

$ sudo -H pip install --upgrade $TF_BINARY_URL

验证安装成功

在终端下执行：

python
import tensorflow

看到类似下图就ok：