如何启用AMD Radeon图形来训练深度学习模型？

Question

我们可以在GPU上训练深度学习模型，我知道如何启用NVIDIA图形来训练深度学习模型，我只想问一下如何在Jupyter Notebook中使用AMD Radeon图形来训练深度学习模型... < / p>

Answer 1

通常，在cuDNN和CUDA的支持下，我们在 Nvidia图形卡中运行深度学习模型。

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

您可以尝试实现此平台（ROCm），但根据某人进行的实验显示，训练阶段和模型性能不如在Nvidia GPU中运行。

为此，您可以使用 google colab 。