scikit-learn是否在我的GPU上运行?

时间:2017-10-01 11:09:47

标签: python scikit-learn gpu nvidia

这个问题与Will scikit-learn utilize GPU?有关,但我不认为提供相同的答案。我正在对Nvidia GPU执行scikit-learn算法而没有错误,因此假设scikit正在底层硬件上运行。由于scikit-learn不是针对GPU执行的,因此什么是使算法能够运行的过程?

例如,我正在使用Gigabyte Nvidia GTX 1060 WF2 3GB GDDR5 PCI-E使用spec:

运行执行scikit-learn算法
1152 NVIDIA CUDA Cores
1582MHz Base/1797MHz Boost Clock (OC Mode) or 1556MHz Base/1771MHz Boost Clock (Gaming Mode)
3GB GDDR5 8008MHz Memory

使用scikit-learn是否有一些未被执行的核心?

更新:

我使用Nvidia docker容器在指定的GPU上运行容器:https://github.com/NVIDIA/nvidia-docker。我在这个容器上安装了scikit,因此在GPU上执行scikit-learn算法?

1 个答案:

答案 0 :(得分:2)

scikit-learn没有也无法在GPU上运行。请参阅scikit-learn常见问题解答中的this answer

相关问题