tensorflow环境搭建

网上找了tensorflow搭建的很多教程,推荐一下tensorflow中文翻译。今天按照里面说的教程搭建,选择了第一种方式pip的形式搭建,并没有搭建成功,遇到一些问题解决方案也较少,在stackoverflow查了一圈,国外很多人推荐采用Virtualenv这种沙箱机制安装,在这期间也遇到一些问题,慢慢解决。

系统环境

tensorflow针对不同的系统,python版本,GPU环境都有不同的下载包,所以需要选择正确的文件安装。
这里选择的是python2.7, Mac OS。
python环境查看

1	$ python --version

安装软件

安装pip和virtualenv环境

1 2	$ sudo easy_install pip # 如果还没有安装 pip $ sudo pip install --upgrade virtualenv

建立一个全新的 virtualenv 环境

为了将环境建在 ~/tensorflow 目录下, 执行:

$ virtualenv --system-site-packages ~/tensorflow
$ cd ~/tensorflow
###然后, 激活 virtualenv:
$ source bin/activate  # 如果使用 bash
(tensorflow)$  # 终端提示符应该发生变化
###在 virtualenv 内, 安装 TensorFlow:
(tensorflow)$ pip install --upgrade $TF_BINARY_URL

测试

在virtualenv 环境下执行测试代码。

$ python
...
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
# 这里一般会显示你的设备显卡信息
>>> print(sess.run(hello))
Hello, TensorFlow!
>>> a = tf.constant(10)
>>> b = tf.constant(32)
>>> print(sess.run(a + b))
42
>>>

至此,环境搭建完成。