How to compile Tensorflow with SSE4.2 and AVX instructions?

前端 未结 12 740
南笙
南笙 2020-11-22 04:14

This is the message received from running a script to check if Tensorflow is working:

I tensorflow/stream_executor/dso_loader.cc:125] successfully opened CUD         


        
12条回答
  •  别那么骄傲
    2020-11-22 04:41

    To hide those warnings, you could do this before your actual code.

    import os
    os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
    import tensorflow as tf
    

提交回复
热议问题