如何在 Google Colab 中获得分配的 GPU 规格

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

这使阅读更容易

!nvidia-smi -L

Answer 3

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