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

我使用Google Colab进行深度学习，我知道他们会随机分配GPU给用户。我希望能够看到我在任何给定会话中分配了哪个GPU。在谷歌Colab笔记本电脑中有这样做的方法吗

---

请注意，如果有帮助的话，我正在使用Tensorflow。

因为您可以在colab中运行bash命令，只需运行

！英伟达smi

：

---

这使阅读更容易

!nvidia-smi -L

---

在collab中运行这两个命令

CUDA：让我们检查一下Nvidia CUDA驱动程序是否已经预装，是哪个版本。

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

!nvidia-smi

---

谢谢英伟达smi的输出是卷积的。这更容易理解。