How to use tensorflow v2 with directml backend

Question

I have a computer with the windows operating system with an amd gpu (rx 5600 xt), and I want to run tensorflow on the gpu. I found "tensorflow-directml" which allows me to run tensorflow on my gpu, but it uses tensorflow 1.14.0. Is there another version of "tensorflow-directml" that uses tensorflow v2, or is there another way to run tensorflow in my gpu?

Thanks, and sorry if I wrote something wrong or inaccurate

Answer 1

Microsoft has announced DirectML-plugin for tensorflow 2 in June this year. Check it out at this link: https://learn.microsoft.com/en-us/windows/ai/directml/gpu-tensorflow-plugin. However I believe for your particular GPU model DirectML-plugin may not be compatible as of yet.