如何将 AMD GPU 用于 fastai/pytorch？

Posted 2023-03-12

【中文标题】如何将 AMD GPU 用于 fastai/pytorch？

【英文标题】：How to use AMD GPU for fastai/pytorch?

【发布时间】：2020-11-10 10:50:12

【问题描述】：

我使用的是配备 Intel Corporation HD Graphics 5500（rev 09）和 AMD Radeon r5 m255 显卡的笔记本电脑。

有谁知道如何设置深度学习，特别是 fastai/Pytorch？

【参考方案1】：

更新 2：

微软在几个小时前发布了Pytorch_DML。 您现在可以使用 pypi package:pytorch-directml 1.8.0a0.dev211021

安装它（在 Windows 或 WSL 中）

pip install pytorch-directml

因此，如果您使用的是 Windows 或使用 WSL，您可以尝试一下！

更新：

从 Pytorch 1.8（2021 年 3 月 4 日）开始，AMD ROCm 版本可从 Pytorch 的官方 website 获得。您现在可以轻松地在 Linux 和 Mac 上安装它们，就像安装 CUDA/CPU 版本一样。

目前，仅提供 pip 包。此外，仍然不支持 Mac 和 Windows 平台（不过我还没有用 WSL2 测试过！）

旧答案：

您需要安装 ROCm 版本。 AMD 关于构建 Pytorch 的官方说明是here。

以前有一个用于 rocm 的***包，但似乎 AMD 不再分发它，相反，您需要从源代码构建 PyTorch，正如我链接到的指南所解释的那样。

但是，您可以参考此页面来构建最新的 PyTorch 版本：The unofficial page of ROCm/PyTorch。

【讨论】：

是否支持Vega 11？我只在该列表中看到 Vega 10。

我不知道，他们的repo似乎有问题，可能值得一看link

【参考方案2】：

更新：2021 年 3 月，Pytorch 增加了对 AMD GPU 的支持，您可以像其他所有基于 CUDA 的 GPU 一样安装和配置它。 Here是链接

不了解 PyTorch，但即使 Keras 现在与 TF 集成，您也可以使用库 PlaidML link 在 AMD GPU 上使用 Keras！英特尔制造的。它非常酷且易于设置，而且为不同项目切换 Keras 后端非常方便