如何控制将Tensorflow CPU功能编译成二进制文件?

时间:2017-06-26 13:59:28

标签: tensorflow

我在运行Tensorflow程序时遇到运行时错误。我跟踪了问题,源代码似乎是我编译的共享对象包含我的处理器不支持的CPU功能。这导致一些指针指向无效地址。目前,tensorflow / core / platform / cpu_info.h中定义了37个CPU功能。我的问题是在编译Tensorflow的源代码时如何不包含其中的一些内容。

1 个答案:

答案 0 :(得分:0)

编译器(./configure)的优化标志有CC_OPT_FLAGS选项。默认情况下,这是-march=native,它将尝试包含当前体系结构支持的指令。您可以使用以下方法手动关闭其中一些-mno-avx关闭AVX。

相关问题