sklearn可以用gpu加速吗?

  统计/机器学习 Python    浏览次数:18798        分享
1

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

 

图不二   2019-02-14 13:50



   2个回答 
6

sklearn不支持GPU

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

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

SofaSofa数据科学社区DS面试题库 DS面经

Nagozi   2019-02-18 06:53

3

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面试题库 DS面经

Zealing   2019-02-24 00:40



  相关讨论

sklearn错误ImportError: cannot import name 'joblib'

python sklearn模型中random_state参数的意义

sklearn训练classifier的时候报错Unknown label type

sklearn模型当中的verbose是什么意思?

查看sklearn版本

sklearn中除了sgd以外,还有什么可以使用partial_fit方法吗?

求sklearn.pipeline.Pipeline使用的例子

python里怎么安装CatBoostClassifier?

CatBoostClassifier里的rsm参量是什么意思?

sklearn里的LabelEncoder什么用?

  随便看看

抛的硬币直到连续出现两次正面为止,平均要扔多少次

xgboost怎么调参?

统计学中的自变量和因变量分别是什么意思?

随机森林会发生过拟合(overfitting)吗?

线性回归需要满足哪些基本前提假设