GPU indicator

Proposed change

Would love an indictor of whether a session has a GPU connected ( kind of like it currently shows the amount of RAM used and Total )

Alternative options

unfortunately right now this is how i would do this

import torch

device = (
    "cuda"
    if torch.cuda.is_available()
    else "mps"
    if torch.backends.mps.is_available()
    else "cpu"
)
print(f"Using device: {device}")

Who would use this feature?

Hubs launching on GPU - make sure you have GPU connected ?

(Optional): Suggest a solution