DeepSpeed
3c490f9c
- Use accelerator to replace cuda in setup and runner (#5769)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Use accelerator to replace cuda in setup and runner (#5769) Use accelerator apis to select device in setup.py and set visible devices env in runner.py --------- Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
References
#5769 - Use accelerator to replace cuda in setup and runner
Author
Andy666G
Parents
249c1db2
Loading