TensorFlow Bazel构建失败

Question

我正在按照TensorFlow的bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer指示使用Bazel构建TensorFlow,并根据来源的说明进行安装.

我收到以下错误:

    ERROR: /home/ubuntu/tensorflow/tensorflow/stream_executor/BUILD:5:1: C++ compilation of rule '//tensorflow/stream_executor:stream_e
xecutor' failed: crosstool_wrapper_driver_is_not_gcc failed: error executing command third_party/gpus/crosstool/clang/bin/crosstool
_wrapper_driver_is_not_gcc -U_FORTIFY_SOURCE '-D_FORTIFY_SOURCE=1' -fstack-protector -fPIE -Wall -Wunused-but-set-parameter -Wno-fr
ee-nonheap-object ... (remaining 87 argument(s) skipped): com.google.devtools.build.lib.shell.BadExitStatusException: Process exite
d with status 1.   

tensorflow/stream_executor/cuda/cuda_dnn.cc: In function 'cudnnConvolutionFwdAlgo_t perftools::gputools::cuda::{anonymous}::ToConvF
orwardAlgo(perftools::gputools::dnn::AlgorithmType)':                                                                              
tensorflow/stream_executor/cuda/cuda_dnn.cc:269:10: error: 'CUDNN_CONVOLUTION_FWD_ALGO_FFT' was not declared in this scope         
     case CUDNN_CONVOLUTION_FWD_ALGO_FFT:  

...

堆栈:运行Ubuntu 14.04.2的EC2 g2.8xlarge机器.Bazel版本0.1.5(已安装w/bazel-0.1.5-jdk7-installer-linux-x86_64.sh).

我尝试过Bazel 0.1.4和0.2.3,我得到了同样的错误.

Answer 1

我在 Ubuntu 16.04 中构建张量流时遇到了同样的问题。

首先确保您使用的 gcc 版本 <= 4.8

就我而言，我必须安装它：

对于海湾合作委员会

sudo apt-get install gcc-4.8
sudo update-alternatives --remove-all gcc
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.8 10

对于 g++

sudo apt-get install g++-4.8
sudo update-alternatives --remove-all g++
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.8 10

一旦有了正确版本的 gcc 和 g++，我就必须编辑 CROSSTOOL 文件，如下所示：

gedit tensorflow_sources_folder/third_party/gpus/crosstool/CROSSTOOL

搜索该特定行的每次出现：

tool_path { name: "gcc" path: "clang/bin/crosstool_wrapper_driver_is_not_gcc" }

并在其正上方插入以下行：

cxx_flag: "-D_FORCE_INLINES"

所以结果一定是：

cxx_flag: "-D_FORCE_INLINES"
tool_path { name: "gcc" path: "clang/bin/crosstool_wrapper_driver_is_not_gcc" }