Qu'est-ce qui est utilisé en cours d'exécution scikit-learn contre GPU?

Question

Cette question est en rapport avec Will scikit-learn utilize GPU? mais je ne pense pas qu'elle offre la même réponse. J'exécute des algorithmes scikit-learn contre un GPU Nvidia sans erreur donc supposons que scikit est en cours d'exécution sur le matériel sous-jacent. Comme scikit-learn n'est pas conçu pour s'exécuter contre GPU, quel est le processus qui permet aux algorithmes de s'exécuter?

Par exemple, je suis en exécutant des algorithmes scikit-learn en utilisant Gigabyte Nvidia GTX 1060 WF2 3GB GDDR5 PCI-E avec spécifications:

1152 NVIDIA CUDA Cores 
1582MHz Base/1797MHz Boost Clock (OC Mode) or 1556MHz Base/1771MHz Boost Clock (Gaming Mode) 
3GB GDDR5 8008MHz Memory 

En utilisant scikit-learn sont quelques-uns des noyaux ne sont pas exécutés contre?

Mise à jour:

J'utiliser le conteneur Nvidia docker pour exécuter conteneur sur GPU comme indiqué: https://github.com/NVIDIA/nvidia-docker. J'ai installé scikit sur ce conteneur afin que les algorithmes scikit-learn soient en cours d'exécution sur GPU?

Answer 1

scikit-learn ne fonctionne pas et ne peut pas fonctionner sur le GPU. Voir this answer dans la FAQ scikit-learn.