scSLAT.utils.get_free_gpu

scSLAT.utils.get_free_gpu()

Get index of GPU with least memory usage

Return type:

int

Ref

https://stackoverflow.com/questions/58216000/get-total-amount-of-free-gpu-memory-and-available-using-pytorch