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

Question

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

例如，我正在运行使用带有规范的Gigabyte Nvidia GTX 1060 WF2 3GB GDDR5 PCI-E执行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，所以scikit-learn算法正在GPU上执行？

Answer 1

scikit-learn不会也不能在GPU上运行。在scikit-learn常见问题解答中查看此答案。

Answer 2

根据我的经验，我使用这个包来利用GPU进行一些sklearn算法。

我使用的代码：

from sklearnex import patch_sklearn
from daal4py.oneapi import sycl_context
patch_sklearn()

来源：英特尔® Scikit-learn 扩展中的 oneAPI 和 GPU 支持