实现算法

现成的模型

无 GPU

sklearn

模型最全, 文档成熟. 但训练大批量的数据就不管用.

有 GPU

cuML

需要带NVIDIA显卡能翻墙的 Linux Host。

• 小爷我还没用过, 因为搞这个需要重写部分代码, 还要装个双系统.

• 这东西正在开发. 文档是没sklearn 成熟. 它们的自我介绍摘抄一段cuML enables data scientists, researchers, and software engineers to run traditional tabular ML tasks on GPUs without going into the details of CUDA programming. In most cases, cuML's Python API matches the API from scikit-learn.

scikit-cuda

模型比较全，文档成熟。

github上打秋风

除了cuML上有，其它找不到，都是写论文打打嘴炮。

• Talk is cheap. Show me the code

需要自己写模型

无 GPU

写个屁

有GPU

tensorflow

pytorch

RAPIDS

cuML 是用这个写的.

跑算法

待尝试

• github workflow (不讲武德，白嫖，慢慢去跑吧)