感觉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面经