web-dev-qa-db-fra.com

Comment obtenir les spécifications GPU attribuées dans Google Colab

J'utilise Google Colab pour une profonde apprentissage et je suis conscient qu'ils allouent au hasard des GPU aux utilisateurs. J'aimerais pouvoir voir quel GPU que j'ai été alloué à une session donnée. Existe-t-il un moyen de le faire dans Google Colab Keatbooks?

Notez que j'utilise Tensorflow si cela aide.

8
Alexander Soare

Puisque vous pouvez exécuter une commande bash à Colab, il suffit de courir !nvidia-smi: enter image description here

5
Vladimir Sotnikov

Cela facilite la lecture plus facile

!nvidia-smi -L
1
korakot

Exécutez ces deux commandes en collab

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

!nvidia-smi
1
Akash Desai