我目前正在使用带有AMD Radeon显卡的联想Ideapad PC。我试图使用卷积神经网络运行图像分类器模型。数据集包含50000个图像,训练模型需要很长时间。有人能告诉我如何使用我的AMD GPU来加固这个过程。我认为AMD Graphics不支持CUDA。那有什么办法吗?
PS:我正在使用Ubuntu 17.10
答案 0 :(得分:1)
您要求的是OpenCL支持,或者更宏大的术语:加速设备的民主化。似乎暂时支持OpenCL,我看到有些人在2018年初测试它,但它还没有完全出现。这个问题在这里被追踪了很长时间:
https://github.com/tensorflow/tensorflow/issues/22
您还应该了解XLA的开发,尝试通过LLVM(或类似LLVM)虚拟化层虚拟化张量流,使其更具可移植性。它目前被引用为2018年初的alpha版。
https://www.tensorflow.org/performance/xla/
目前还没有一个简单的解决方案,但这些是遵循这些方针的两项努力。