centos - 如何安装 tensorflow 显卡？

Question

我目前在安装带有 gpu 支持的 tensorflow 时遇到一些问题。

这是我遵循的指南。

安装 NVIDIA CUDA(预装)

安装 NVIDIA cuDNN(预装)

安装 bazel

wget https://github.com/bazelbuild/bazel/releases/download/0.4.3/bazel-0.4.3-installer-linux-x86_64.sh      
chmod +x bazel-0.4.3-installer-linux-x86_64.sh
./bazel-0.4.3-installer-linux-x86_64.sh --user

从源安装 tensorflow

git 克隆 https://github.com/tensorflow/tensorflow

cd tensorflow /

。/配置

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

最后一个没有完成..或者它完成了..它出现了一条错误消息

a@fe1:~/tensorflow$ bazel build -c opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

.
WARNING: Sandboxed execution is not supported on your system and thus hermeticity of actions cannot be guaranteed. See http://bazel.build/docs/bazel-user-manual.html#sandboxing for more information. You can turn off this warning via --ignore_unsupported_sandboxing.

Server finished RPC without an explicit exit code

在这之后我应该能够做到这一点

bazel-bin/tensorflow/tools/pip_package/build_pip_package /tmp/tenso
rflow_pkg

根据指南，但不可能。

我之前运行过tensor flow cpu版本，但由于对gpu的需求非常迫切，我决定安装它......但没想到会这么麻烦..

关于为什么无法构建它的任何想法？

操作系统:centOS 7.1
显卡:英伟达

Answer 1

我在尝试使用 Bazel 从源代码安装 Tensorflow 时也遇到了问题。这可能是当前的 bazel 版本有问题，因为我过去曾发生过这种情况。如果您向我们提供有关您的系统(操作系统)的信息，这将有所帮助＆GPU)，但你可能最好使用pip(或pip3)并执行sudo pip(3)安装tensorflow-gpu。