Google Tensorflow是否支持OpenCL

时间:2017-04-28 01:02:13

标签: tensorflow opencl

Google Tensorflow是否支持OpenCL ......还是只有Cuda?

OpenCL似乎尚未得到支持(2017年4月)每个开放的问题 - https://github.com/tensorflow/tensorflow/issues/22但我一直在阅读支持存在 - 我可能会遗漏一些东西。

2 个答案:

答案 0 :(得分:16)

tf-coriander是针对OpenCL 1.2 GPU的Tensorflow实现。它基于coriander,它是在OpenCL 1.2设备上运行NVIDIA®CUDA™代码的通用编译器。披露:我是这两个项目的作者。

答案 1 :(得分:4)

在Tensorflow上通过SYCL提供OpenCL支持,有些功能在,其他功能正在进行中:https://github.com/tensorflow/tensorflow/issues/22#issuecomment-266050835