-1

我已到达 Tensorflow 0.9 教程的设置 TensorFlow 以进行开发部分,但在尝试添加对 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/core/kernels/batchtospace_op_gpu.cu.cc' 包含的以下文件

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

设置:
基本操作系统 0.3
tensorflow 0.9
cuda-7.5
cudnn 5.0.5
gcc 4.8.4

4

1 回答 1

0

尝试添加

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

third_party/gpus/crosstool/CROSSTOOL

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

于 2016-07-21T19:39:19.833 回答