AMD & Microsoft Collaborate To Bring TensorFlow-DirectML To Life, Up To 4.4x Improvement on RDNA 2 GPUs
AMD ROCm 4.5 finally brings support for RDNA2 GPUs. Good news for Tensorflow, Pytorch and HIP developers. : r/radeon
They say that AMD does not support machine learning...Today I got TensorFlow up and running on my 6900 XT (using tensorflow-directml) and started fine-tuning GPT2. Pictured below: what my gpu looks like. :
ROCm™: Machine Learning | AMD
Accelerating TensorFlow Performance on Mac — The TensorFlow Blog
Community supported AMD ROCm build for TensorFlow | by TensorFlow | TensorFlow | Medium
GitHub - ROCmSoftwarePlatform/tensorflow-upstream: TensorFlow ROCm port