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

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

    This is the simplest method. Only one step.

    It has significant impact on speed. In my case, time taken for a training step almost halved.

    Refer custom builds of tensorflow

提交回复
热议问题