有没有办法（或者可能是列表？）来确定所有提供GPU支持的tensorflow操作？ 现在，这对我来说是一个试错过程 - 我尝试在GPU上进行一组操作。如果它的工作，甜蜜。如果没有，请尝试其他地方。
到目前为止，这是我发现的唯一相关（但没有帮助）：https://github.com/tensorflow/tensorflow/issues/2502
答案 0 :(得分：1)
不幸的是，似乎没有内置的方法来获取确切的列表，甚至没有检查特定操作。如上面的问题所述，调度是在本机C ++代码中完成的：如果相应的内核已注册到DEVICE_GPU，则可以将特定操作分配给GPU。
我认为最简单的方法是grep "REGISTER_KERNEL_BUILDER" -r tensorflow张量流源码来获取匹配操作列表，这些操作看起来像like this。
但请记住，即使使用REGISTER_KERNEL_BUILDER规范，也无法保证在GPU上执行操作。例如，无论现有内核如何，都会在CPU上分配32位int Add。