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

Question

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

Answer 1

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

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

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

作为此问题的解决方案，您可以使用 google colab。