感觉sklearn还是有点慢的,特别是gridsearch调参的时候。sklearn可以用gpu加速吗?
2个回答
sklearn不支持GPU
截止到2019年2月,官网上依然标明了“在可预见的未来,sklearn不会考虑增加gpu支持。”
原因以及原文看链接:will sklearn add gpu support
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面经 相关主题
sklearn SGDClassifier的partial_fit是什么意思?
3回答
sklearn cross_val_score怎么同时对多个scoring进行验证
2回答
sklearn训练classifier的时候报错Unknown label type
1回答
sklearn里learning_rate什么意思?
2回答
sklearn.model_selection.cross_val_predict怎么固定random_state?
1回答
我们谢绝在回答前讲“生动”的故事。
我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。
我们谢绝答非所问。
我们谢绝自己不会、硬要回答。
我们感激每一个用户在编写答案时的努力与付出!