构建pip时使用GPU支持构建错误

时间:2016-07-21 18:45:58

标签: tensorflow bazel

我已经到达了Tensorflow 0.9教程的Setting up TensorFlow for Development部分,我在尝试添加对GPU的支持时遇到了错误。

\> bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package   
  

错误:/ home /.../tensorflow/tensorflow/core/kernels/BUILD:1489:1:   规则中未声明的包含   ' // tensorflow / core / kernels:batchtospace_op_gpu':缺少此规则   包含的以下文件的依赖声明   ' tensorflow /型芯/核/ batchtospace_op_gpu.cu.cc'

但是,如果我排除--config=cuda指令,我不会收到此错误。

设定:
基本OS 0.3
tensorflow 0.9
CUDA-7.5
cudnn 5.0.5
gcc 4.8.4

1 个答案:

答案 0 :(得分:0)

尝试添加

cxx_builtin_include_directory: "/usr/local/cuda-8.0/include"

third_party/gpus/crosstool/CROSSTOOL

https://github.com/tensorflow/tensorflow/issues/3431