安装TensorFlow的方法有很多种,根据不同的平台,不同的方式,有着不一样的安装方法,详见官方指导:
https://www.tensorflow.org/install/install_linux#ValidateYourInstallation
这里主要记录Ubuntu下搭建一个独立的虚拟环境来运行TensorFlow,这种方法的优点是,独立一个目录,有着自己的python和tensorflow环境,不与其它冲突,也就是说你可以用这种方式安装多个版本,而互不冲突。
主要步骤:
1、安装python环境:详见 https://www.tensorflow.org/install/install_linux#InstallingVirtualenv
安装后,记得运行环境脚本:source 安装目录/bin/activate
退出环境后,再进入,都要重新运行脚本。
2、启动完第1步的环境后,输入安装tensorflow的命令:
pip install --upgrade TF_PYTHON_URL
其中的 TF_PYTHON_URL 是tensorflow的python包链接,可根据自己的安装要求进行选择安装。各种tensorflow的python包可到此链接下载:
https://www.tensorflow.org/install/install_linux#the_url_of_the_tensorflow_python_package
3、验证安装是否成功。主要方法是启动python环境,敲入一段简单的代码:
$python
>>>import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>>print(sess.run(hello))
打印出: Hello, TensorFlow! 就表示成功