|
|
|
|
|
by rolleiflex
1619 days ago
|
|
Yes, they have the Radeon Open Compute project (ROCM) but they seem to be intent on following the tensorflow from a few stable versions behind. Additionally, and likely this is not something they can do anything about, but if you are using a Linux VM instead of running it natively, ROCM does not work. I had attempted to do this to get some use of the AMD GPU that was in my Mac — no dice. The holy grail would be a direct replacement backend that could be fed into TF, like CUDA. |
|
ROCm has a direct replacement backend that can even take CUDA code (it's designed to be incredibly similar). It's called HIP. It's just that no one wants to support it. That is actually how TensorFlow on AMD works (mostly), and you can compile the latest stable release that way.