5 ms·
I posted this because I have noticed many people aren't aware TensorFlow can run on AMD gpus. When informed, a common response is how well does it work? This G
by montalbano 7y ago
I posted this because I have noticed many people aren't aware TensorFlow can run on AMD gpus. When informed, a common response is how well does it work?
This GitHub issue provides answers to both those questions with respect to the latest major release of TensorFlow. (Admittedly just a single benchmark data point, but better than nothing.)
- paol 7y agoFrom skimming the thread I get the impression it's a beta quality / work-in-progres kind of deal. Is that the case?
- edgyquant 7y agoThat's because it really doesn't work on AMD gpus (at least on Linux.). Sure if you spent a lot of time setting up the rocm driver you might get lucky but it's alpha quality at best and in my case (and the case of several people in this thread and the GitHub one linked) it doesn't work at all.
- acoye 7y agoI was following the old GH issue / thread related to OpenCL support for 1.x of TensorFlow. I tinkered an got it working with opensource kernel drivers on GCN1 (HD 7970) https://github.com/tensorflow/tensorflow/issues/22 https://github.com/tensorflow/tensorflow/issues/22 You could do it, and GCN was good at compute (still is with Radeon VI) but I must admit it was far more complicated than any AI researcher would have liked compared to the closed source binary approach of NVidia / Cuda.