DeepSpeed
add available memory check to accelerators
#4508
Merged

add available memory check to accelerators #4508

tjruwase merged 6 commits into master from jeffra/available_memory
jeffra
jeffra add available memory check to accelerator
a251e592
jeffra catch case where nvmlInit fails
567ba717
jeffra add pynvml to reqs
bc16f063
jeffra jeffra requested a review from mrwyattii mrwyattii 1 year ago
jeffra jeffra requested a review from tjruwase tjruwase 1 year ago
jeffra jeffra requested a review from cmikeh2 cmikeh2 1 year ago
jeffra fix for cpu systems
26c2bcd3
mrwyattii
mrwyattii commented on 2023-10-12
jeffra Update accelerator/cuda_accelerator.py
d7b0764d
jeffra simplify
6a747d6f
tjruwase
tjruwase
tjruwase approved these changes on 2023-10-12
mrwyattii
mrwyattii approved these changes on 2023-10-13
delock
tjruwase tjruwase merged 12aedac6 into master 1 year ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone