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.

score 55 · Accepted Answer · answered Feb 19 '20 at 12:06

55

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

answered Feb 19 '20 at 12:06

Vladimir Sotnikov

1,399
11
13

> In Colab, you can invoke shell commands using either ! or %%shell. – Alex78191 Apr 13 '22 at 18:48

score 36 · Answer 2 · answered Feb 20 '20 at 00:04

36

This makes it easier to read

!nvidia-smi -L

answered Feb 20 '20 at 00:04

korakot

37,818
16
123
144

Thanks! The output of `nvidia-smi` was convoluted. This is so much easier to understand. – Harshit Trehan Apr 18 '21 at 09:24

score 4 · Answer 3 · answered Sep 27 '20 at 06:02

4

Run this two commands in collab

CUDA: Let's check that Nvidia CUDA drivers are already pre-installed and which version is it.

!/usr/local/cuda/bin/nvcc --version

!nvidia-smi

answered Sep 27 '20 at 06:02

Akash Desai

498
5
11

How to get allocated GPU spec in Google Colab

3 Answers3