检查TensorFlow安装是否具有XLA支持

Question

如何检查是否使用XLA支持构建了特定的TensorFlow安装？ configure中有一个选项可以关闭或打开它，但我不知道如何从Python中查询这些选项。

Answer 1

根据版本1.12.0的发行说明

  

1.12的预构建二进制文件默认情况下具有XLA支持链接。

您应该可以在里面检查它 tensorflow.contrib.compiler

或简单地运行：

import tensorflow as tf
from tensorflow.contrib.compiler import xla

如果存在ImportError，则表明xla不存在。 从here引用的代码。

另请参考XLA Overview。