我目前正在使用带有 AMD Radeon 显卡的联想 Ideapad PC。我正在尝试使用卷积神经网络运行图像分类器模型。数据集包含 50000 张图像，训练模型花费的时间太长。有人可以告诉我如何使用我的 AMD GPU 来加快进程。我认为 AMD 显卡不支持 CUDA。那么有什么办法吗？
PS：我使用的是 Ubuntu 17.10
问问题
1241 次
1 回答
2
您要求的是 OpenCL 支持，或者用更宏大的术语来说：加速设备的民主化。似乎对 OpenCL 有初步的支持，我看到有人在 2018 年初对其进行了测试，但它似乎还没有完全成熟。该问题已在此处跟踪了很长时间：
https://github.com/tensorflow/tensorflow/issues/22
您还应该了解 XLA 上的开发，这是一种尝试在 LLVM（或类似 LLVM）的虚拟化层上虚拟化 tensorflow，使其更具可移植性。它目前被引用为截至 2018 年初的 alpha 版本。
https://www.tensorflow.org/performance/xla/
目前还没有一个简单的解决方案，但这是遵循这些思路的两项努力。
于 2018-04-21T14:58:47.613 回答