Python 在GPU上安装tensorflow

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

这取决于您的图形卡，它必须是nvidia，并且您必须在您的系统上安装相应的cuda版本等等。然后，在已安装的CUDA版本上安装相应的cuDNN

步骤：

安装NVIDIA 367驱动程序

安装CUDA 8.0

安装cuDNN 5.0

重新启动

使用上述配置从bazel的源位置安装tensorflow

---

您的图形卡不支持CUDA驱动程序，如果没有CUDA驱动程序，您将无法在GPU上使用tensorflow。您的系统将运行tensorflow，但仅在CPU上运行

不过，您可以使用pytorch，这是完成类似任务的另一种方法。PyTorch有另一个名为CLTorch的版本，它运行在图形卡上的OpenCL上

请点击此链接了解更多详细信息。

---

首先，如果您希望看到性能提升，您应该拥有更好的GPU，其次，Tensorflow使用CUDA，这仅适用于具有3.0或更高CUDA功能的NVidia GPU。如果你真的想进行深度学习，我建议你使用一些云服务，如AWS或Google cloud。

如果你想使用带有TensorFlow的AMD GPU，你可以按照说明操作

然而：

• 您使用的GPU没有那么强大，不太可能给您带来很大的性能提升
• 尽管有Windows版本的ComputeCpp，但还没有使用TensorFlow对其进行测试，但您需要在这些说明中使用Linux

---

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