如果我有AMD显卡,如何使用tensorflow-gpu版本?

时间:2018-10-11 12:01:22

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

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

2 个答案:

答案 0 :(得分:1)

据我所知,您需要Linux和ROCm才能拥有支持Tensorflow的AMD GPU。 对于ubuntu和ROCm 1.9.1,在Python3中:

nameof

答案 1 :(得分:0)

Tensflow不仅可以在CPU上运行,而且还可以在CPU上运行,因此只要满足依赖关系,它就可以在所有系统上运行。但是,我认为您的问题是关于让GPU使用Tensorflow进行计算,而Tensorflow可能是tensorflowGPU版本。您可以做几件事。

  1. 使用Linux和ROCm作为替代。 -ROCm早在几年前就发布了,以使您想要的一切变得可能。可能会有些麻烦,性能下降约5-10%,但仍然比任何Tensorflow CPU实现都要好得多。

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

  1. 使用TensorflowJS:如果您不太喜欢使用python,则可以在所有支持opengl的机器上使用Tensorflow.js。我没有使用过,但是我已经阅读了一些。请仔细阅读文档