tensorflow的安装有两种方式:
①简单安装直接pip:
pip install tensorflow
②源码安装:
详见本文内容
在pip安装时,需要提前安装好对应的CUDA、CUDNN。比如tensorflow1.0需要CUDA8.0版本。由于我事先使用caffe已经安装了CUDA7.5,CUDNN 5.0。如果安装CUDA8.0,会影响caffe等的使用,因此这里考虑tensorflow的源码安装。
环境:
Ubuntu14.04
1 下载tensorflow源码
$git clone https://github.com/tensorflow/tensorflow
2 安装编译工具bazel
官网安装文档:https://bazel.build/versions/master/docs/install.html
其中给出了三种安装方法,这里介绍第一种方法:传统的API库方法。
①安装JDK8
事先已经安装,此处省略,文末的参考博客可供参考
②将bazel加入到源
$ echo “deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8” | sudo tee /etc/apt/sources.