如何将 AMD GPU 用于 fastai/pytorch？

Question

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

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

Answer 1

更新 ：

从 Pytorch 1.8（2021 年 3 月 4 日）开始，AMD ROCm 版本可从 Pytorch 的官方网站获得。您现在可以轻松地在 Linux和 Mac上安装它们，与安装 CUDA/CPU 版本的方式相同。

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

旧答案：

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

以前有一个用于 rocm 的轮包，但 AMD 似乎不再分发它了，相反，您需要从源代码构建 PyTorch 作为我链接到的指南进行解释。

但是，您可以参考此页面来构建最新的 PyTorch 版本：ROCm/PyTorch 的非官方页面。

Answer 2

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

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