我正在关注本教程安装与 CUDA Compute Capability 3.0 兼容的支持 GPU 的 TensorFlow。
我安装了 Java-JDK8、Bazel 0.1.0、TensorFlow 0.6.0,并更改了配置以在 CUDA Compute Capability 3.0 上运行。到目前为止一切都很好。
但是当我输入这个命令时:
$HOME/bin/bazel build -c opt --config=cuda //tensorflow/cc:tutorials_example_trainer
我看到这个输出:
Extracting Bazel installation...
.....
ERROR: /home/me/tensorflow/tensorflow/core/BUILD:1: Extension file not found: 'google/protobuf/protobuf.bzl'.
ERROR: /home/me/tensorflow/tensorflow/cc/BUILD:65:1: error loading package 'tensorflow/core': Extension file not found: 'google/protobuf/protobuf.bzl' and referenced by '//tensorflow/cc:tutorials_example_trainer'.
ERROR: Loading failed; build aborted.
INFO: Elapsed time: 1.006s
有什么建议吗?
答案1
运行以下命令可修复该问题:
$ git clone -b 0.6.0 –recurse-子模块https://github.com/tensorflow/tensorflow.git
非常感谢@steeldriver建议。