AMD Pytorch ROCM support #8
Description
@crowsonkb has informed me this should work but I can't currently test it as I don't have an AMD GPU.
I'd greatly appreciate it if someone with that type of setup could let me know what their experience is using this codebase.