欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

tensorflow 环境搭建

程序员文章站 2022-03-29 19:29:20
...

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