diff --git a/tortoise/utils/device.py b/tortoise/utils/device.py index 9c35962..3ab52e2 100755 --- a/tortoise/utils/device.py +++ b/tortoise/utils/device.py @@ -18,7 +18,7 @@ def set_device_name(name): def get_device_name(): global DEVICE_OVERRIDE - if DEVICE_OVERRIDE is not None: + if DEVICE_OVERRIDE is not None and DEVICE_OVERRIDE != "": return DEVICE_OVERRIDE name = 'cpu'