我的印象是TF仅支持CUDA。根据我的研究,使其获得支持OpenCL和AMD GPU的唯一方法是获得一个单独的“适配器”模块,例如TF Coriander,并遭受性能故障的困扰:
从GitHub:
对于循环网络,Coriander比直接使用NVIDIA®CUDA™慢约4倍。
检查“洞察”选项卡,该模块的确似乎处于不断发展的状态,但是我不确定该如何乐观。因此,我是否应该对此感到乐观,并期望在不到几年的时间里获得接近本地的经验?如果不是,是否计划为实际的本机解决方案提供支持?如果不是,AMD即将面世的7nm架构是否会改变开发者的想法?
我意识到这可能有点题外话。如果是,请告诉我是否可以放其他地方。
作为contrib的一部分,您可以使用SYCL支持构建Tensorflow。
SYCL是“单源OpenCL”,这是Khronos的一项新标准,它允许编写高级C ++代码,这些代码可以编译为在OpenCL设备上运行。
CodePlay软件的人们参与其中很多,您可以在此处看到他们关于该主题的博客文章。
简而言之,您不会获得其中的点子包,而是需要自己构建。由于项目仍处于初期阶段,因此性能可能不如之前。
您可以在此处找到有关如何入门的教程。请记住,这使用了CodePlay的SYCL宣传版,但也许您可以将其与开放式实现(例如triSYCL)一起使用。