bytedance / bytedance/InfiniteYou

Multi-GPU support ?

Open
#9 2 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
2.7k
Forks
285
PR merge metrics
No merged PRs in 30d

Description

Hi,

I have 11 GPU in my server but the script don't spread across gpu, which end with an out of memory error.

Any multi-gpu support in the future ?

Contributor guide

No contributing guide indexed for this repository

Research direction

The issue names no files, tests, or entry points. Start by locating the script's GPU selection and model-loading paths, then reproduce the out-of-memory failure on the reported multi-GPU setup. Done means the workload is distributed across available GPUs without the reported memory failure and the behavior is covered by an appropriate test or documented usage.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
distributed-systems, machine-learning, performance
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.