How to run ImageAI on AMD GPU?

[ImaHoomaan]

I am using RX 5700 XT. when training the model, it only runs on my CPU and not on my GPU.

[OlafenwaMoses]

Will put this on the list of updates to add next..

• https://pytorch.org/blog/pytorch-for-amd-rocm-platform-now-available-as-python-package/
• https://learn.microsoft.com/en-us/windows/ai/directml/gpu-pytorch-windows