Since Pytorch for Radeon GPUs is available, how to put data on device

since there is the ROCm 4.0 version provided from PyTorch, which (if I am not wrong) provides the interfaces for other GPUs than nvidia, how can I select my gpu as device?
E.g. device = torch.device(“cuda”) does not work with radeons.
I want to know, how I can put the e.g. CNN onto the GPU. I am glad for any help.