这个问题与scikit-learn 会使用 GPU 吗?但我认为没有提供相同的答案。我正在针对 Nvidia GPU 执行 scikit-learn 算法,没有错误,因此假设 scikit 正在底层硬件上运行。由于 scikit-learn 并非设计用于针对 GPU 执行,那么支持算法运行的进程是什么?
例如,我正在使用以下命令运行 scikit-learn 算法Gigabyte Nvidia GTX 1060 WF2 3GB GDDR5 PCI-E
规格:
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 上执行?
scikit-learn 不能也不可能在 GPU 上运行。看这个答案在 scikit-learn 常见问题解答中。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)