微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

GPU/TPU 上的 PyCaret 方法

如何解决GPU/TPU 上的 PyCaret 方法

CPU Utilization

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

解决方法

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

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。