tensorflow 环境搭建

google 人工智能框架工具搭建

对cpu和gpu有要求,详情见文档 我以我mac为例

  • 首先要安装python 2.7 或者3.x 没有或者版本低可以升级
$ sudo easy_install pip
 $ sudo pip install --upgrade virtualenv
  • 创建环境目录,官网建议(~/tensorflow)就是你的家目录下面新建一个tensorflow文件夹
$ virtualenv --system-site-packages targetDirectory
  • 根据你所使用的sh环境选择激活命令
$ source ~/tensorflow/bin/activate      # If using bash, sh, ksh, or zsh
 $ source ~/tensorflow/bin/activate.csh  # If using csh or tcsh

成功之后应该有提示

 (tensorflow)$

  • 如果你的pip版本是8.1以及以后版本,选择安装语句,建议使用第一条
$ pip install --upgrade tensorflow      # for Python 2.7
 $ pip3 install --upgrade tensorflow     # for Python 3.n
 $ pip install --upgrade tensorflow-gpu  # for Python 2.7 and GPU
 $ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU

否则操作如下(备选方案)

$ pip install --upgrade TF_BINARY_URL   # Python 2.7
 $ pip3 install --upgrade TF_BINARY_URL  # Python 3.N

TF_BINARY_URL 如下,一般为了避免麻烦不会使用备选方案

 $ pip3 install --upgrade \
 https://storage.googleapis.com/tensorflow/mac/cpu/tensorflow-1.0.0-py3-none-any.whl

  • 测试环境
$ source ~/tensorflow/bin/activate      # If using bash, sh, ksh, or zsh

Python 测试

$ python

>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))

退出

(tensorflow)$ deactivate

  • 卸载
$ rm -r ~/tensorflow

你可能感兴趣的:(tensorflow 环境搭建)