How to get allocated GPU spec in Google Colab

Question

I'm using Google Colab for deep learning and I'm aware that they randomly allocate GPU's to users. I'd like to be able to see which GPU I've been allocated in any given session. Is there a way to do this in Google Colab notebooks?

Note that I am using Tensorflow if that helps.

Answer 1

Since you can run bash command in colab, just run !nvidia-smi:

Answer 2

This makes it easier to read

!nvidia-smi -L