参见官网 https://www.tensorflow.org/install/install_sources
ubuntu16.04,python3.5,cuda8.0,cudnn 5.1.10
以上条件都已满足。
1. install GNU coreutils
sudo apt-get install coreutils
2. 安装bazel
参见官网 https://bazel.build/versions/master/docs/install.html
第一步
sudo add-apt-repository ppa:webupd8team.java
sudo apt-get update
sudo apt-get install oracle-java8-installer
第二步
echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | sudo tee /etc/apt/sources.list.d/bazel.list
curl https://bazel.build/bazel-release.pub.gpg | sudo apt-key add -
第三步
sudo apt-get update && sudo apt-get install bazel
sudo apt-get upgrade bazel
3. Configure the installation
下载tensorflow,进入目录进行配置
git clone --recurse-submodules https://github.com/tensorflow/tensorflow
cd tensorflow
./configure
按照提示做完即可,别填错自己的python路径,gcc路径和cuda以及cudnn路径,不然后面编译可能会失败。
4. build a pip package with gpu
bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package