How to make pytorch recognize AMD GPU

I am using an AMD R9 390.
I Installed pytorch given the instructions from the following suggestions:

However in python torch.cuda.is_availible() returns false. As I understood, this ROCm version is meant to use the amd gpu as cuda.

If there are additional steps I need to take to utilize the GPU, please let me know. Thank you.