recommenders-team/recommenders

XDeepFM not utilizing GPU [ASK]

Open

#1.900 aberto em 16 de mar. de 2023

Ver no GitHub
 (0 comments) (0 reactions) (0 assignees)Python (2.972 forks)batch import
help wanted

Métricas do repositório

Stars
 (17.706 stars)
Métricas de merge de PR
 (Mesclagem média 6d 16h) (10 fundiu PRs em 30d)

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

Guia do colaborador

Direção de pesquisa
Verifique se o modelo e os dados estão na GPU examinando a alocação de dispositivo. Use ferramentas de perfil como nvidia smi e time do Python para identificar se o carregamento de dados vinculado à CPU é o gargalo. Certifique se de que as operações tensoriais sejam executadas na GPU.
Pilha de tecnologia
python
Domain
machine learningbackend
Tipo Issue
Bug
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-3 horas
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Principalmente claro
Prerequisites
Pythondeep learning basicsGPU training concepts
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
50