那么有没有办法在 Windows 上使用 AMD 的 GPU 来加速 PyTorch 呢?答案是肯定的,多谢 DirectML 和 Torch-DirectML。安装步骤请参考 Enable PyTorch with DirectML on Windows | Microsoft Learn。值得一提的是官方文档提及仅支持 PyTorch 1.13,但我测试 PyTorch 2.0.0