如何在 AMD/ATI GPU 上运行 TensorFlow？

读完本教程后https://www.tensorflow.org/guide/using_gpu https://www.tensorflow.org/guide/using_gpu我用这个简单的代码检查了 GPU 会话

import numpy as np
import matplotlib.pyplot as plt
import tensorflow as tf

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2,3], name = 'a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape = [3,2], name =  'b')
c = tf.matmul(a, b)

with tf.Session(config=tf.ConfigProto(log_device_placement=True)) as sess:
    x = sess.run(c)
print(x)

输出是

2018-08-07 18:44:59.019144: I
tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports
instructions that this TensorFlow binary was not compiled to use: AVX2
FMA Device mapping: no known devices. 2018-08-07 18:44:59.019536: I
tensorflow/core/common_runtime/direct_session.cc:288] Device mapping:

MatMul: (MatMul): /job:localhost/replica:0/task:0/device:CPU:0
2018-08-07 18:44:59.019902: I
tensorflow/core/common_runtime/placer.cc:886] MatMul:
(MatMul)/job:localhost/replica:0/task:0/device:CPU:0 a: (Const):
/job:localhost/replica:0/task:0/device:CPU:0 2018-08-07
18:44:59.019926: I tensorflow/core/common_runtime/placer.cc:886] a:
(Const)/job:localhost/replica:0/task:0/device:CPU:0 b: (Const):
/job:localhost/replica:0/task:0/device:CPU:0 2018-08-07
18:44:59.019934: I tensorflow/core/common_runtime/placer.cc:886] b:
(Const)/job:localhost/replica:0/task:0/device:CPU:0 [[ 22.  28.]  [
49.  64.]]

如您所见，GPU 没有进行任何计算。 当我更改代码以使用 GPU 的配置和进程分数时：

conf = tf.ConfigProto()
conf.gpu_options.per_process_gpu_memory_fraction = 0.4

with tf.Session(config = conf) as sess:
    x = sess.run(c)
print(x)

输出是

2018-08-07 18:52:22.681221: I
tensorflow/core/platform/cpu_feature_guard.cc:141] Your CPU supports
instructions that this TensorFlow binary was not compiled to use: AVX2
FMA [[ 22.  28.]  [ 49.  64.]]

我该如何在 GPU 卡上运行会话？

在 AMD GPU 上运行 TensorFlow 是绝对可能的。大约2年前ROCm被发布，事情就完成了。然而，需要注意的是，由于其开源起源，它目前只能在 Linux 上运行。因此，如果您愿意使用 Linux，那么您肯定可以使用 AMD GPU 来训练您的深度学习模型。也就是说，由于社区仍然不够大，您将获得的支持量很低。 Google 搜索 ROCm，您可以获得有关如何在 Linux 计算机上设置和运行它的说明。也许它可以在 Windows 中与 WSL2 一起使用，但我还没有尝试过，所以无法对此发表评论。

这是 ROCm 安装文档的链接 https://rocmdocs.amd.com/en/latest/Installation_Guide/Installation-Guide.html