utils.devices.try_gpu

Contents

utils.devices.try_gpu#

try_gpu(i: int = 0) device | None[source]#

Return gpu(i) if exists, otherwise return None