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

Ale*_*are 18 python gpu tensorflow google-colaboratory

我正在使用 Google Colab 进行深度学习,我知道他们会随机将 GPU 分配给用户。我希望能够查看在任何给定会话中分配给我的 GPU。有没有办法在 Google Colab 笔记本中做到这一点?

请注意,如果有帮助,我正在使用 Tensorflow。

Vla*_*kov 24

由于您可以在 colab 中运行 bash 命令,只需运行!nvidia-smi在此处输入图片说明


Kor*_*ich 12

这使阅读更容易

!nvidia-smi -L
Run Code Online (Sandbox Code Playgroud)