TF for AMD CPU on Windows

[fsergeyal]

Hello,

I have AMD Ryzen Threadripper CPU but performance is not so good. I'm using Tensorflow with MKL. As I understand it is better to use Tensorflow with OpenBLAS

Do you have any plans to build TF for AMD CPUs? If not, Any advice how can I build this by myself?

Thanks in advance! Sergey.

[Fannhhyy]

Maybe you can try: os.environ["MKL_DEBUG_CPU_TYPE"] = "5" I test 3950x , tf-mkl not work well

[fsergeyal]

Thanks for your comment!

It helped to dig futher.

I found two articles:

Here is performance comparison with os.environ["MKL_DEBUG_CPU_TYPE"] = "5" https://www.pugetsystems.com/labs/hpc/How-To-Use-MKL-with-AMD-Ryzen-and-Threadripper-CPU-s-Effectively-for-Python-Numpy-And-Other-Applications-1637/

And here is comparison using AMD's BLIS library: https://www.pugetsystems.com/labs/hpc/AMD-Threadripper-3970x-Compute-Performance-Linpack-and-NAMD-1631/

So, I seems that BLIS library is the best way to use.

Do you have ideas how to get Tensorflow works with BLIS library instead of MKL?

Thanks.