TensorFlow安装(借助virtualenv)

virtualenv可以为python配置不同的环境(包依赖,python版本等), 借助virtualenv安装tensorflow,可以方便使用不同的python版本(例如python2, python3), 也可以方便的卸载(直接将env删除)

安装

前置条件:安装了python/virtualenv/python-pip

1. 为tensorflow创建一个env
   virtualenv tensorflow
   目录下将出现一个叫tensorflow的目录,目录下bin/lib等一应俱全,这就是virtualenv为我们准备的单独环境
2. 切换到该环境
   source tensorflow/bin/activate
3. 安装tensorflow
   pip install --upgrade tensorflow
   稍等片刻, tensorflow即安装成功了.

Hello World

尝试运行以下Python代码:

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')

session = tf.Session()
print session.run(hello)


你可能会遇到类似以下警告:
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.

这是由于直接用pip安装的是兼容各种cpu的版本, 并不支持你的CPU的一些高级特性. 如果你想提高性能,需要从源码编译

这条警告可以通过加入以下代码移除:

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'


卸载

将virtualenv为你创建的目录删除即可
rm -rf tensorflow