Windows 11搭配AMD GPU的PyTorch安装求助

核心问题解答 #

• CUDA toolkit无法替代ROCm：CUDA是NVIDIA专属的并行计算框架，仅适配NVIDIA GPU，AMD设备无法通过CUDA实现PyTorch的GPU加速。
• 是否需要更换Linux系统？：如果想让AMD GPU发挥PyTorch的加速能力，切换到Linux（如Ubuntu）并搭配ROCm是目前官方支持的唯一成熟方案，ROCm对AMD GPU的兼容性和优化更完善。
• 无需更换系统/使用CUDA的替代方法：
  • 安装CPU版本的PyTorch：直接使用PyTorch的CPU发行版，虽无GPU加速，但能正常运行所有PyTorch代码，适合小规模模型调试或非算力密集型场景。
  • 尝试PyTorch的DirectML后端：微软与AMD合作推出的DirectML支持Windows平台的AMD GPU加速PyTorch，安装步骤如下：

    pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
    pip install torch-directml
    

    使用时在代码开头添加设备指定：

    import torch_directml
    device = torch_directml.device()
    

    该方案的性能和兼容性略逊于Linux+ROCm，但能满足基础GPU加速需求。

内容的提问来源于stack exchange，提问作者burplePixel