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

前端 未结 12 742
南笙
南笙 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:50

    I compiled a small Bash script for Mac (easily can be ported to Linux) to retrieve all CPU features and apply some of them to build TF. Im on TF master and use kinda often (couple times in a month).

    https://gist.github.com/venik/9ba962c8b301b0e21f99884cbd35082f

提交回复
热议问题