TensorFlow安装(借助virtualenv)
2017-06-14 本文已影响0人
吃土的汉子
virtualenv可以为python配置不同的环境(包依赖,python版本等), 借助virtualenv安装tensorflow,可以方便使用不同的python版本(例如python2, python3), 也可以方便的卸载(直接将env删除)
安装
前置条件:安装了python/virtualenv/python-pip
- 为tensorflow创建一个env
virtualenv tensorflow
目录下将出现一个叫tensorflow的目录,目录下bin/lib等一应俱全,这就是virtualenv为我们准备的单独环境 - 切换到该环境
source tensorflow/bin/activate - 安装tensorflow
pip install --upgrade tensorflow
稍等片刻, tensorflow即安装成功了.
Hello World
尝试运行以下Python代码:
<code>
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
session = tf.Session()
print session.run(hello)
</code>
你可能会遇到类似以下警告:
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的一些高级特性. 如果你想提高性能,需要从源码编译
这条警告可以通过加入以下代码移除:
<code>
import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
</code>
卸载
将virtualenv为你创建的目录删除即可
rm -rf tensorflow