GPU/TPU 上的 PyCaret 方法

问题描述

当我运行 TOTAL 时，cpu 内存负载很大，而我的 GPU 未使用。如何在 GPU 上运行 best_model = compare_models() 或 setup()？ PyCaret 中是否有内置方法？

解决方法

要在 PyCaret 中使用 gpu，您必须简单地将 use_gpu=True 作为设置函数中的参数。 例子： 模型 = 设置（数据，target_variable，use_gpu=True）