PyTorch ROCm 已出 - 如何将 select Radeon GPU 作为设备

Question

since Pytorch released the ROCm version, which enables me to use other gpus than nvidias, how can I select my radeon gpu as device in python? 显然，像 device = torch.cuda.is_available 或 device = torch.device("cuda") 这样的代码不起作用。 谢谢你的帮助。 :)

Answer 1

这让我花了很长时间才弄清楚。 我的 AMD GPU 仍然存在一些配置问题，所以我无法测试它是否有效，但是，根据这个 github pytorch 线程，设备 Rocm 集成是编写的，所以你可以cuda') 并且不需要实际的移植！

请参阅此处的线程： https://github.com/pytorch/pytorch/issues/10670