使用AMD培训深度学习模型

时间:2018-04-21 14:40:32

标签: tensorflow deep-learning amd-processor convolutional-neural-network

我目前正在使用带有AMD Radeon显卡的联想Ideapad PC。我试图使用卷积神经网络运行图像分类器模型。数据集包含50000个图像,训练模型需要很长时间。有人能告诉我如何使用我的AMD GPU来加固这个过程。我认为AMD Graphics不支持CUDA。那有什么办法吗?

PS:我正在使用Ubuntu 17.10

1 个答案:

答案 0 :(得分:1)

您要求的是OpenCL支持,或者更宏大的术语:加速设备的民主化。似乎暂时支持OpenCL,我看到有些人在2018年初测试它,但它还没有完全出现。这个问题在这里被追踪了很长时间:

https://github.com/tensorflow/tensorflow/issues/22

您还应该了解XLA的开发,尝试通过LLVM(或类似LLVM)虚拟化层虚拟化张量流,使其更具可移植性。它目前被引用为2018年初的alpha版。

https://www.tensorflow.org/performance/xla/

目前还没有一个简单的解决方案,但这些是遵循这些方针的两项努力。