recommenders-team/recommenders

XDeepFM not utilizing GPU [ASK]

Open

#1,900 opened on 2023年3月16日

GitHub で見る
 (0 comments) (0 reactions) (0 assignees)Python (2,972 forks)batch import
help wanted

Repository metrics

Stars
 (17,706 stars)
PR merge metrics
 (平均マージ 6d 16h) (30d で 10 merged PRs)

説明

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

コントリビューターガイド