Run nvidia-smi in notebook cell that keeps updating while other cells do their work

nvidia-smi.py

from datetime import timedelta
import multiprocessing
import subprocess
import time

from IPython.display import clear_output


def nvidia_smi_call():
    """
    Call `nvidia-smi` in the background and refresh the cell output with the stdout every second
    The cell where this is run in will not block and other cells can be run after it while it keeps updating.
    """
    start_time = time.time()
    count = 0
    while True:
        clear_output(wait=True)
        result = subprocess.run(["nvidia-smi"], stdout=subprocess.PIPE)
        time_delta = timedelta(seconds=time.time() - start_time)
        print(
            result.stdout.decode("utf-8") + "\n"
            + f"Run for: {time_delta!s}s ({count} updates)"
        )
        time.sleep(1)
        count += 1


nvidia_smi_proc = multiprocessing.Process(target=nvidia_smi_call)
nvidia_smi_proc.start()