sklearn可以用gpu加速吗?

  统计/机器学习 Python    浏览次数:287        分享
0

感觉sklearn还是有点慢的,特别是gridsearch调参的时候。sklearn可以用gpu加速吗?

 

图不二   2019-02-14 13:50



   2个回答 
5

sklearn不支持GPU

截止到2019年2月,官网上依然标明了“在可预见的未来,sklearn不会考虑增加gpu支持。”

原因以及原文看链接:will sklearn add gpu support

SofaSofa数据科学社区 DS面经 问答 实战

Nagozi   2019-02-18 06:53

2

Sklearn本身不支持gpu。但有个变通方法,在tensorflow上运行sklearn,让TF去支持GPU运算。有个skflow,

https://github.com/tensorflow/tensorflow/tree/master/tensorflow/contrib/learn/python/learn

可以试试。

sklearn不会包含GPU compiler, 而TF其实是像Matlab一样的计算平台,包含对操作系统和GPU compiler的支持。

SofaSofa数据科学社区 DS面经 问答 实战

Zealing   2019-02-24 00:40



  相关主题

sklearn实现adaboost算法了吗?   1回答

GradientBoostingClassifier的每次输出结果不同   1回答

查看sklearn版本   1回答

sklearn.linear_model.SGDClassifier中的参数n_iter设置问题   2回答

sklearn的模型如何保存下来?   2回答

python sklearn模型中random_state参数的意义   2回答

sklearn cross_val_score中的参数pre_dispatch   1回答

lightgbm.LGBMModel和lightgbm.LGBMClassifier该用哪个?   1回答

sklearn.linear_model.SGDClassifier能做小批量学习(mini-batch)吗?   2回答

sklearn模块有神经网络的类可以调用吗?   1回答

sklearn predict的默认阈值   2回答

sklearn里LogisticRegressionCV中的参量Cs什么意思   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!