recommenders-team/recommenders

XDeepFM not utilizing GPU [ASK]

Open

#1 900 ouverte le 16 mars 2023

Voir sur GitHub
 (0 commentaires) (0 réactions) (0 assignés)Python (2 972 forks)batch import
help wanted

Métriques du dépôt

Stars
 (17 706 stars)
Métriques de merge PR
 (Merge moyen 6j 16h) (10 PRs mergées en 30 j)

Description

Description

I'm currently implementing XDeepFM but can't get training on the gpu. I followed the instructions in the GPU support section here.

When I trigger the training, I see the GPU memory in use but utilization stays @ 0. Sometimes, it changes to 1% and then 6% but the majority of the time, it stays @ 0%.

A single cpu stays at 100% however. I do not believe the GPU is being utilized. Any ideas?

Other Comments

Guide contributeur

Direction de recherche
Vérifiez si le modèle et les données sont sur le GPU en examinant l'affectation du périphérique. Utilisez des outils de profilage comme nvidia smi et time de Python pour identifier si le chargement des données lié au CPU est le goulot d'étranglement. Assurez vous que les opérations tensorielles sont exécutées sur le GPU.
Stack technique
python
Domaine
machine learningbackend
Type d'issue
Bug
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-3 heures
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Plutôt claire
Prérequis
Pythondeep learning basicsGPU training concepts
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
50