Recommended way to use AMD GPU with PyTorch ROCM?

Using PyTorch we are able to access AMD GPU by specifying device as ‘cuda’.
Is this the recommended way to access AMD GPU through PyTorch ROCM?

1 Like