TensorFlow是否计划支持OpenCL?

时间:2019-01-23 23:34:49

标签: tensorflow opencl

我觉得TF仅支持CUDA。根据我的研究,使其获得支持OpenCL和AMD GPU的唯一方法是获得一个单独的“适配器”模块,例如TF Coriander并遭受性能故障的困扰:

Performance Chart

从GitHub:

  

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

查看“洞察”选项卡,该模块似乎确实处于持续开发中,但是我不确定该如何乐观。因此,我是否应该对此感到乐观,并期望在不到几年的时间里获得接近本地的经验?如果不是,是否计划为实际上本机解决方案提供支持?如果不是,AMD即将面世的7nm架构是否会改变开发人员的想法?

我意识到这可能有点题外话。如果是这样,请告诉我是否可以放其他地方。

1 个答案:

答案 0 :(得分:1)

作为contrib的一部分,您可以使用SYCL支持构建Tensorflow。

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

CodePlay软件的人们参与其中很多,您可以在主题here上看到他们的博客文章。

因此,简而言之,您不会获得其中的点子包,而是需要自己构建。而且,由于该项目仍处于初期阶段,因此性能可能不会很好。

您可以找到有关here入门的教程。请记住,这使用了CodePlay的SYCL的推广版本,但也许您可以将其与开放式实现(例如triSYCL)一起使用。