Get memory info for all GPUs in Numba

gpumem.py

from numba import cuda

gpus = cuda.gpus.lst

for gpu in gpus:
    with gpu:
        meminfo = cuda.current_context().get_memory_info()
        print("%s, free: %s bytes, total, %s bytes" % (gpu, meminfo[0], meminfo[1]))