recommenders-team/recommenders

XDeepFM not utilizing GPU [ASK]

Open

#1900 aperta il 16 mar 2023

Vedi su GitHub
 (0 commenti) (0 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

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

Guida contributor

Direzione di ricerca
Controlla se il modello e i dati sono sulla GPU esaminando l'allocazione del dispositivo. Utilizza strumenti di profilazione come nvidia smi e time di Python per identificare se il caricamento dei dati legato alla CPU è il collo di bottiglia. Assicurati che le operazioni tensoriali siano eseguite sulla GPU.
Tech stack
python
Dominio
machine learningbackend
Tipo issue
Bug
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Abbastanza chiara
Prerequisiti
Pythondeep learning basicsGPU training concepts
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
50