一、安装前的准备工作
在正式开始安装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!”就代表安装成功了。