如何安装张量流显卡?

Posted

技术标签:

【中文标题】如何安装张量流显卡?【英文标题】:How do install tensor flow gpu? 【发布时间】:2017-06-07 07:59:51 【问题描述】:

我目前在安装支持 gpu 的张量流时遇到一些问题。

这是我遵循的指南。

安装 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

从源安装张量流

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

cd张量流/

./配置

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 显卡:英伟达

【问题讨论】:

【参考方案1】:

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

【讨论】:

pip install tensor flow-gpu 对我的不起作用,这就是为什么我首先要这样做 你使用的是哪个操作系统和python版本? 您确定 Red Hat 衍生产品支持 gpu 版本吗? 我正在使用外部集群,因为它是为深度学习目的而构建的。我认为一切都应该没问题..

以上是关于如何安装张量流显卡?的主要内容,如果未能解决你的问题,请参考以下文章

如何在我的代码中使用张量核心而不是 cuda 核心?

我们可以在 Mac 上拥有多个张量流版本吗?

无法安装张量流

安装张量流时出错

这个张量流消息是啥意思?有啥副作用吗?安装成功了吗?

如何将张量流张量切成多个