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

大概意思是大概意思是:你的CPU支持AVX扩展,但是你安装的TensorFlow版本无法编译使用。

也就是你安装的tensorflow版本太low了

  (1) 如果安装的是GPU版本

        如果你有一个GPU,你不应该关心AVX的支持,因为大多数昂贵的操作将被分派到一个GPU设备上(除非明确地设置)。在这种情况下,您可以简单地忽略此警告:

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

   (2) 如果安装的是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性能也应该改善。

 而由于我安装的是gpu版本,所以直接忽略掉就可以了

参考警告:Your CPU supports instructions that this TensorFlow binary was not compiled to use: AVX2 FMA_feng98ren的专栏-CSDN博客

posted @ 2022-05-15 21:49  贪睡地蜗牛  阅读(48)  评论(0编辑  收藏  举报