sccross.models.utils.autodevice

sccross.models.utils.autodevice()[source]

Get torch computation device automatically based on GPU availability and memory usage

Return type

device

Returns

device – Computation device