NVIDIA vs AMD for tensorflow

As mentioned in the z440 post, the workstation comes with a NVIDIA Quadro K5200. I've been a happy user of AMD hardware since Radeon HD 4850 (upgraded 5870 and R9 390 later). Unfortunately, tensorflow only supports Cuda - possibly due to missing OpenCL support in Eigen. Is it worth switching just for that? I did a few experiments:

CPU (8core Xeon E5-1680v3)
real 8m24.991s
user 104m13.048s
sys  0m38.772s

All CPUs ~80% usage

GPU (NVIDIA Quadro K5200)
real 2m12.990s
user 2m47.136s
sys  0m30.500s

All CPUs ~20% usage

I think it's worth it - about 4x improvement by using the GPU vs a high end 8 core Xeon. And this GPU is 2 generations back - a GTX 1080 or newer will probably give an even higher benefit.
So for now, I'll stick with NVIDIA.

Popular posts from this blog

HP z440