recommenders-team/recommenders

[ASK] Utilizing GPU with LightGCN Pipeline

Open

#1650 aperta il 18 feb 2022

Vedi su GitHub
 (0 commenti) (2 reazioni) (0 assegnatari)Python (2972 fork)batch import
help wanted

Metriche repository

Star
 (17.706 star)
Metriche merge PR
 (Merge medio 6g 16h) (10 PR mergiate in 30 g)

Descrizione

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!

Guida contributor

Direzione di ricerca
Controlla se PyTorch/TensorFlow riconosce la GPU eseguendo torch.cuda.is available() o tf.config.list physical devices('GPU'). Assicurati che il modello sia spostato sulla GPU con .to(device) o device='/gpu:0'.
Tech stack
python
Dominio
backend
Tipo issue
Documentazione
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
2
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
Meno di 1 ora
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
Python
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
75