keras - 如何启用 AMD Radeon 显卡来训练深度学习模型？

Question

我们可以在 GPU 上训练我们的深度学习模型，我知道如何启用 NVIDIA 显卡来训练深度学习模型，我只想问我们如何使用 AMD Radeon 显卡在 Jupyter Notebook 中训练深度学习模型......

标签： kerasdeep-learningjupyter-notebookamdtraining-data

Answer 1

通常，我们在Nvidia 显卡上运行深度学习模型是在cuDNN 和 CUDA 的支持下进行的。

据我所知，基于 AMD 显卡的 ROCm 现在支持 TensorFlow、Caffe、MXnet 等。

你可以尝试实现这个平台（ROCm），但是根据有人运行的实验表明，训练阶段和模型性能不如在 Nvidia GPU 上运行。

作为解决方案，您可以使用google colab。