【发布时间】：2021-01-06 18:22:13
【问题描述】：
我们可以在 GPU 上训练我们的深度学习模型，我知道如何启用 NVIDIA 显卡来训练深度学习模型，我只想问我们如何使用 AMD Radeon 显卡在 Jupyter Notebook 中训练深度学习模型...
【问题讨论】：
标签： keras deep-learning jupyter-notebook amd training-data
【发布时间】：2021-01-06 18:22:13
【问题描述】：
通常，我们在 Nvidia 显卡 中运行深度学习模型，并得到 cuDNN 和 CUDA 的支持。
据我所知，基于 AMD 显卡的 ROCm 现在支持 TensorFlow、Caffe、MXnet 等。
你可以尝试实现这个平台（ROCm），但是根据有人运行的实验表明，训练阶段和模型性能不如在Nvidia GPU上运行。
作为解决方案，您可以使用 google colab。
【讨论】：