scikit-learn会利用GPU吗？
10 浏览
scikit-learn会利用GPU吗？
阅读在TensorFlow中实现scikit-learn：http://learningtensorflow.com/lesson6/和scikit-learn：http://scikit-learn.org/stable/modules/generated/sklearn.cluster.KMeans.html。我很难决定使用哪种实现方式。
scikit-learn作为tensorflow docker容器的一部分安装，因此可以使用任一实现方式。
使用scikit-learn的原因：
scikit-learn比tensorflow的实现方式少了一些样板代码。
使用tensorflow的原因：
如果在Nvidia GPU上运行，算法将并行运行，我不确定scikit-learn是否会利用所有可用的GPU？
阅读https://www.quora.com/What-are-the-main-differences-between-TensorFlow-and-SciKit-Learn
TensorFlow更低级，基本上是帮助您实现机器学习算法的乐高积木，而scikit-learn为您提供现成的算法，例如分类算法（如支持向量机，随机森林，逻辑回归等）。如果要实现深度学习算法，TensorFlow非常出色，因为它允许您利用GPU进行更高效的训练。
这个陈述加强了我的观点，“scikit-learn比tensorflow的实现方式少了一些样板代码”，但也暗示scikit-learn不会利用所有可用的GPU。