在 GPU 上安装 tensorflow [重复]答案

【问题标题】：Installing tensorflow on GPU [duplicate]在 GPU 上安装 tensorflow [重复]
【发布时间】：2018-02-25 13:17:27
【问题描述】：

我已经安装了 tensorflow CPU 版本。我使用的是 Windows 10，并且我的 GPU 为 AMD Radeon 8600M。我现在可以安装 GPU 版本的 tensorflow 吗？会不会有什么问题？如果没有，我在哪里可以获得安装 GPU 版本的说明？

【问题讨论】：

有点宽泛的问题，所以我会给出一些宽泛的建议。您应该能够使用 pip install 安装 tensorflow-gpu。您还需要 CUDA（代码可以在没有它的情况下运行，但不会使用 GPU）。 TensorFlow 有一个 CIFAR-10 GPU 示例教程。
请参阅this list 以查看您的 GPU 是否受 CUDA 支持。您需要 3.0 或更高的计算能力，TF 才能在 GPU 上工作

【解决方案1】：

您的显卡不支持 CUDA 驱动程序，否则您无法在 GPU 上使用 tensorflow。您的系统将运行 tensorflow，但只能在 CPU 上运行。

但是您可以使用 pytorch，这是完成类似任务的另一种方式。 PyTorch 有另一个名为 CLTorch 的版本，它在 OpenCL 上运行，而 OpenCL 在你的显卡上运行。

请点击此链接了解更多详情。 https://github.com/hughperkins/cltorch

【讨论】：

【解决方案2】：

首先，如果你想看到性能提升，你应该有更好的 GPU，其次，Tensorflow 使用 CUDA，这仅适用于 CUDA Capability 为 3.0 或更高的 NVidia GPU。如果你真的想做深度学习，我建议你使用一些云服务，比如 AWS 或 Google Cloud。

【讨论】：

【解决方案3】：

如果您想在 TensorFlow 中使用 AMD GPU，可以按照说明进行操作 here。

但是：

【讨论】：

【解决方案4】：

这取决于你的显卡，它必须是nvidia，并且你必须在你的系统和SO上安装对应的cuda版本。然后，您已经安装了与您已安装的 CUDA 版本相对应的 cuDNN

步骤：

【讨论】：