TensorFlow GPU支持Mac - OpenCL

Question

TensorFlow是否支持在运行AMD Radeon R9 M370X的2015年末款mac上使用GPU？

AMD Radeon R9 M370X:
Chipset Model:  AMD Radeon R9 M370X
Type:   GPU
Bus:    PCIe
PCIe Lane Width:    x8
VRAM (Total):   2048 MB
Vendor: ATI (0x1002)
Device ID:  0x6821
Revision ID:    0x0083
ROM Revision:   113-C5670E-777
Automatic Graphics Switching:   Supported
gMux Version:   4.0.20 [3.2.8]
EFI Driver Version: 01.00.777
Metal:  Supported

- user3554141

---

可能是使用Keras和Tensorflow与AMD GPU的重复问题。 - Steven

MacBook上没有对AMD的支持（可能永远不会有）。但是在Linux上，对AMD还是有一点希望的。 - Yaroslav Bulatov

Answer 1

目前不支持OpenCL。

您可以在此处查看进展情况：https://github.com/tensorflow/tensorflow/issues/22

- Steven

---

SYCL实现使得在OpenCL设备上运行TensorFlow的最新更新已经到来，正在取得进展。详情请参见https://www.codeplay.com/portal/02-14-17-bringing-the-acceleration-of-opencl-to-tensorflow-with-sycl - Rod Burns

Answer 2

好消息！tf-coriander 支持使用 AMD GPU 运行 TensorFlow。已在 MBP2017 AMD Radeon 560 上验证通过！！

- Gianmar

---

但是TensorFlow版本太旧了，可能比仅使用CPU运行更慢。 - wyx

该代码库目前存在问题。在本文撰写时（1.11），它无法与最新的Tensorflow版本兼容。 - rayryeng

Answer 3

如果你使用的是AMD显卡，你可以使用OpenCL来完成任务，而NVIDIA则可以轻松地使用CUDA。我已经测试过了，它可以正常工作。你可以在这里找到完整的步骤。https://dev59.com/LlIH5IYBdhLWcg3wWck0#60016869

- bikram