您的位置:

安装TensorFlow

一、安装前的准备工作

在正式开始安装TensorFlow之前,需要完成一些准备工作。

1、安装Python环境,并且版本要在2.7.x或3.x之间,推荐使用Python3版本;

python --version

2、在Python环境中安装pip包管理工具,可以使用系统包管理器或者官网提供的安装脚本进行安装;

sudo apt-get install python-pip

3、安装OpenSSL开发包,这个是TLS/SSL的实现,默认情况下,TensorFlow会使用TLS/SSL进行数据传输,所以需要安装OpenSSL来支持;

sudo apt-get install libssl-dev

二、使用pip安装TensorFlow

安装TensorFlow的最简单方便的方式是使用pip命令安装,该方法可用于Linux、Mac OS X和Windows等操作系统。

1、在终端或者命令行下输入以下pip命令进行安装:

pip install tensorflow

2、根据需要选择不同版本的TensorFlow,例如,要安装CPU版本的TensorFlow,可以使用以下命令:

pip install tensorflow-cpu

如果想要安装GPU版本的TensorFlow,需要安装相应的GPU版本的cuDNN和CUDA。

三、使用源代码安装TensorFlow

对于有一定编程基础和一些运维经验的用户,可以选择从源代码编译安装TensorFlow。

1、首先,从GitHub上下载TensorFlow源代码:

git clone https://github.com/tensorflow/tensorflow

2、安装必要的依赖包,包括Python依赖、Bazel构建系统依赖、协议缓冲区编译器等:

sudo apt-get install python-numpy python-dev python-pip python-wheel
sudo apt-get install bazel
sudo apt-get install protobuf-compiler

3、切换到tensorflow目录下,使用bazel构建系统编译TensorFlow:

cd tensorflow
./configure
bazel build --config=opt //tensorflow/tools/pip_package:build_pip_package
bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tensorflow_pkg
sudo pip install /tmp/tensorflow_pkg/tensorflow-version-tags.whl

四、安装完毕后的测试

安装完TensorFlow之后,可以使用Python的交互式界面或者编写Python脚本进行测试。

1、在Python交互式界面中,输入以下代码,看是否可以正常导入tensorflow库:

import tensorflow as tf

2、编写一个简单的脚本文件test.py,内容如下:

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

在终端或者命令行下运行该脚本文件:

python test.py

如果输出了“Hello, TensorFlow!”就代表安装成功了。