recommenders-team/recommenders

XDeepFM not utilizing GPU [ASK]

Open

#1,900 建立於 2023年3月16日

在 GitHub 查看
 (0 留言) (0 反應) (0 負責人)Python (2,972 fork)batch import
help wanted

倉庫指標

Star
 (17,706 star)
PR 合併指標
 (平均合併 6天 16小時) (30 天內合併 10 個 PR)

描述

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

貢獻者指南