使用 AMD 训练深度学习模型

Train Deep learning Models with AMD

我目前正在使用配备 AMD Radeon 显卡的 Lenovo Ideapad PC。我正在尝试运行使用卷积神经网络的图像分类器模型。数据集包含 50000 张图像，训练模型的时间过长。谁能告诉我如何使用我的 AMD GPU 来加快进程。我认为 AMD 显卡不支持 CUDA。那么有什么办法吗？

PS: 我正在使用 Ubuntu 17.10

您要求的是 OpenCL 支持，或者更宏大的说法：加速设备的民主化。似乎有对 OpenCL 的暂定支持，我看到一些人在 2018 年初对其进行了测试，但它似乎还没有完全成熟。这个问题已经在这里跟踪了很长一段时间：

您还应该了解 XLA 上的开发，尝试在 LLVM（或类似 LLVM）的虚拟化层上虚拟化 tensorflow，使其更便携。它目前被引用为截至 2018 年初处于 alpha 阶段。

目前还没有一个简单的解决方案，但这是沿着这些思路进行的两项努力。