美文网首页
Your CPU supports instructions t

Your CPU supports instructions t

作者: MaTae | 来源:发表于2018-11-02 16:37 被阅读0次

    Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2

    如果安装的是CPU版本(pip install tensorflow)

    1.在代码中加入如下代码,忽略警告:

    import os

    os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

    2.编译TensorFlow源码

    如果您没有GPU并且希望尽可能多地利用CPU,那么如果您的CPU支持AVX,AVX2和FMA,则应该从针对CPU优化的源构建tensorflow。在这个问题中已经讨论过这个问题,也是这个GitHub问题。 Tensorflow使用称为bazel的ad-hoc构建系统,构建它并不是那么简单,但肯定是可行的。在此之后,不仅警告消失,tensorflow性能也应该改善。

    相关文章

      网友评论

          本文标题:Your CPU supports instructions t

          本文链接:https://www.haomeiwen.com/subject/veycxqtx.html