How to get allocated GPU spec in Google Colab

python tensorflow gpu google-colaboratory
Alexander Soare picture Alexander Soare · Feb 19, 2020 · Viewed 19.4k times · Source

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

Vladimir Sotnikov picture Vladimir Sotnikov · Feb 19, 2020

Since you can run bash command in colab, just run !nvidia-smi: enter image description here

Related questions

How to tell if tensorflow is using gpu acceleration from inside python shell?

I have installed tensorflow in my ubuntu 16.04 using the second answer here with ubuntu's builtin apt cuda installation. Now my question is how can I test if tensorflow is really using gpu? I have a gtx 960m gpu. When I …

Read More
How to get current available GPUs in tensorflow?

I have a plan to use distributed TensorFlow, and I saw TensorFlow can use GPUs for training and testing. In a cluster environment, each machine could have 0 or 1 or more GPUs, and I want to run my TensorFlow graph into …

Read More
How do I use TensorFlow GPU?

How do I use TensorFlow GPU version instead of CPU version in Python 3.6 x64? import tensorflow as tf Python is using my CPU for calculations. I can notice it because I have an error: Your CPU supports instructions that this …

Read More