DeepSpeed
3c490f9c - Use accelerator to replace cuda in setup and runner (#5769)

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>
Author
Parents
Loading