如果我有 AMD 显卡，如何使用 tensorflow-gpu 版本？答案

【问题标题】：How can I use tensorflow-gpu version if I have AMD graphics card?如果我有 AMD 显卡，如何使用 tensorflow-gpu 版本？
【发布时间】：2018-10-11 12:01:22
【问题描述】：

我在 Windows 10 中使用 Anaconda，但无法安装当前版本的 tensorflow-gpu。我有 4 GB AMD Radeon 显卡。那么如何在笔记本电脑上运行Tensorflow-gpu 版本呢？

【问题讨论】：

How to run TensorFlow on AMD/ATI GPU?的可能重复

标签： python-3.x tensorflow image-processing anaconda object-detection

【解决方案1】：

据我所知，您需要 Linux 和 ROCm 才能拥有支持 Tensorflow 的 AMD GPU。在我使用 ubuntu 和 ROCm 1.9.1 的情况下，在 Python3 中：

import tensorflow as tf; print(tf.__version__)
>>1.11.0

【讨论】：

【解决方案2】：

Tensflow 不仅适用于 CPU，它也适用于 CPU，因此只要满足依赖关系，它将适用于所有系统。但是，我认为您的问题是关于让您的 GPU 使用 Tensorflow 进行计算，这将是 tensorflowGPU 版本。您可以做几件事。

使用 Linux 和 ROCm 作为替代方案。 - ROCm 是几年前发布的，旨在使您想要的一切成为可能。这可能会有点麻烦，性能会下降 5-10%，但它仍然比任何 tensorflow 的 CPU 实现要好得多。

You can visit this link for installation and set up details for ROCm

使用 TensorflowJS：如果你对使用 python 不是很讲究，那么你可以在所有支持 opengl 的机器上使用 Tensorflow.js。我没有使用它，但我读过一些关于它的信息。请浏览文档

【讨论】：