recommenders-team/recommenders

[ASK] Utilizing GPU with LightGCN Pipeline

Open

#1 650 ouverte le 18 févr. 2022

Voir sur GitHub
 (0 commentaires) (2 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

Is there anything that needs to be done to make sure LightGCN is utilizing GPU's? I'm following this deep dive notebook here: https://github.com/microsoft/recommenders/blob/main/examples/02_model_collaborative_filtering/lightgcn_deep_dive.ipynb

My dataset is 1.4M users with about 13M interactions, a single epoch is taking about an hour. I have access to 4 GPU's, 192GB memory.

Thanks!

Guide contributeur

Direction de recherche
Vérifiez si PyTorch/TensorFlow reconnaît le GPU en exécutant torch.cuda.is available() ou tf.config.list physical devices('GPU'). Assurez vous que le modèle est déplacé vers le GPU avec .to(device) ou device='/gpu:0'.
Stack technique
python
Domaine
backend
Type d'issue
Documentation
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
2
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
Moins d'une heure
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.
Claire
Prérequis
Python
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
75