Function available_gpus

dali/test/python/test_utils_tensorflow.py:48–52

()

def available_gpus():
    devices = []
    for device_id in range(num_available_gpus()):
        devices.append("/gpu:{0}".format(device_id))
    return devices


@contextmanager