TensorFlow 是否计划支持 OpenCL？

Question

我的印象是 TF 只支持 CUDA。根据我的研究，让它支持 OpenCL 和 AMD GPU 的唯一方法是获得一个单独的“适配器”模块，例如 TF Coriander 并遭受性能故障：

来自 GitHub：

对于循环网络，Coriander 比直接使用 NVIDIA® CUDA™ 慢 4 倍左右。

该模块似乎在持续开发中，查看洞察选项卡，但我不确定该有多乐观。那么我是否应该对此感到乐观并期待在几年内获得近乎原生的体验？如果没有，是否计划支持实际上原生解决方案？如果不是这样，AMD 即将推出的 7nm 架构会改变开发者的想法吗？

我意识到这可能有点跑题了。如果是，请告诉我是否可以放在其他任何地方。

Answer 1

作为 contrib 的一部分，您可以构建带有 SYCL 支持的 Tensorflow。

SYCL 是“单源 OpenCL”，它是 Khronos 的一项新标准，允许人们编写可编译以在 OpenCL 设备上运行的高级 C++ 代码。

CodePlay 软件的人们已经大量参与其中，您可以查看他们关于主题 here 的博文。

简而言之，你不会得到它的 pip 包，你需要自己构建它。由于该项目仍处于早期阶段，因此性能可能不会那么好。

您可以找到有关如何开始使用的教程here。请记住，这使用了 CodePlay 的 SYCL 支持版本，但也许您可以使用诸如 triSYCL 之类的开放实现来使其工作。