recommenders-team/recommenders

[ASK] Utilizing GPU with LightGCN Pipeline

Open

#1,650 创建于 2022年2月18日

在 GitHub 查看
 (0 评论) (2 反应) (0 负责人)Python (2,972 fork)batch import
help wanted

仓库指标

Star
 (17,706 star)
PR 合并指标
 (平均合并 6天 16小时) (30 天内合并 10 个 PR)

描述

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!

贡献者指南

研究方向
通过运行 torch.cuda.is available() 或 tf.config.list physical devices('GPU') 检查 PyTorch/TensorFlow 是否识别 GPU。确保模型通过 .to(device) 或 device='/gpu:0' 移动到 GPU。
技术栈
python
领域
backend
议题类型
文档
难度面向新贡献者的预计实现难度，1 表示很小改动，5 表示专家级工作。
2
预计时间有经验贡献者完成调查、实现、测试并准备 pull request 的粗略时间范围。
1 小时以内
活动状态议题当前的可参与程度：新鲜、活跃、陈旧、阻塞或等待维护者输入。
活跃
清晰度议题是否清楚说明期望改动、验收标准和下一步。
清晰
前置要求
Python
新手友好度1-100 的估计分数，表示该议题对首次贡献者的友好程度。
75