根据官方文档,PyTorch现在支持AMD GPU。ROCm 4.2可以通过pip进行安装。但是我在谷歌和官方文档中找不到如何强制使用GPU进行深度学习训练的方法。以下命令在AMD上的等效命令是什么?
torch.device('cuda' if torch.cuda.is_available() else 'cpu')
用户名或电子邮箱地址
密码