机器学习案例2:使用 plaidML在 macOS利用GPU进行机器学习

想要在 Mac 的集成 AMD GPU 或外部显卡上训练机器学习模型? PlaidML 就是您的最佳选择。

任何尝试过在 macOS 上使用 TensorFlow 训练神经网络的人都知道这个过程有点糟糕。 TensorFlow 只能利用 Mac 上的 CPU,因为 GPU 加速训练需要 Nvidia 芯片组。大多数大型模型在 CPU 上的训练时间比在简单 GPU 上训练的时间要长几个数量级。

更糟糕的是,许多 Mac 都配备了功能强大的 AMD GPU,这些 GPU 在训练时被迫闲置。 TensorFlow 仅支持 Nvidia 设备,与 macOS 不兼容。这时就是 plaidML 的用武之地。

您可以安装 PlaidML 并使用它直接在 Mac 的GPU图形处理器上训练 Keras 模型,而不是在基于云的系统上花费时间或购买新机器。

你可能感兴趣的:(实用机器学习指南,macos,机器学习,人工智能)