我已经到达了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
答案 0 :(得分:0)
尝试添加
cxx_builtin_include_directory: "/usr/local/cuda-8.0/include"
到
third_party/gpus/crosstool/CROSSTOOL