PR #732 Allow for GPU-ID specification on CLI

Allow for GPU-ID specification on CLI #732

muellerzr merged 9 commits into main from gpu-ids

Specifiy GPU ids on CLI

2d9b5339

Configurable gpu-ids

a38e5eed

Expand to deepspeed

18f55460

all

539d4a84

muellerzr added enhancement

muellerzr requested a review from

pacman100 3 years ago

muellerzr requested a review from

sgugger 3 years ago

muellerzr marked this pull request as draft 3 years ago

sgugger approved these changes on 2022-09-30

Fix nit

2f819213

Fix typo in docs

f93ba9d8

muellerzr marked this pull request as ready for review 3 years ago

futher tweaks

15853e07

Further tweaks

91be3902

pacman100 commented on 2022-09-30

Change for mps specifically

0467d4ff

muellerzr merged 870a7bad into main 3 years ago

muellerzr deleted the gpu-ids branch 3 years ago

Reviewers

sgugger

pacman100

Assignees

No one assigned

Labels

enhancement

Milestone

No milestone