recommenders-team/recommenders

[ASK] Utilizing GPU with LightGCN Pipeline

Open

#1,650 opened on 2022年2月18日

GitHub で見る
 (0 comments) (2 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

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!

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

調査方針
PyTorch/TensorFlow が GPU を認識しているか確認するには、torch.cuda.is available() または tf.config.list physical devices('GPU') を実行します。モデルを .to(device) または device='/gpu:0' で GPU に移動してください。
技術スタック
python
領域
backend
Issue 種別
ドキュメント
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
2
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
1時間未満
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
アクティブ
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
明確
前提条件
Python
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
75