AMD架构的GPU计算平台

Question

众所周知，尝试训练深度神经网络或基本上任何类型的 AI 模型都需要大量的计算能力。在 CPU 上执行此操作可能需要几天时间，具体取决于项目。

我有一个 AMD GPU，具体来说，GPU 的型号名称是“RX5700”。对于 NVIDIA GPU，有可以轻松用于 GPU 计算的 CUDA，但是，我找不到 AMD 架构的替代方案。我知道有 ROCm，但它只适用于非常特殊的 GPU 系列和 Ubuntu 操作系统。

是否有任何平台可以在 Windows 上使用 RX 系列 GPU 进行 GPU 计算？

感谢您的回答。

Answer 1

当我拥有 Radeon 6700XT 时，我能够利用 Microsoft 的 DirectML。它不如 NVIDIA/CUDA/TensorFlow 等。尽管它仍然比在 CPU 上训练要好得多。虽然它仍然使用旧版本的 TensorFlow，但它的局限性很大。