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

Posted 2023-02-23

技术标签:

【中文标题】如何在 Google Colab 中获得分配的 GPU 规格【英文标题】：How to get allocated GPU spec in Google Colab 【发布时间】：2020-06-03 14:24:00 【问题描述】：

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

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

【问题讨论】：

【参考方案1】：

由于您可以在 colab 中运行 bash 命令，因此只需运行 !nvidia-smi：

【讨论】：

【参考方案2】：

这样更容易阅读

!nvidia-smi -L

【讨论】：

谢谢！ nvidia-smi 的输出很复杂。这很容易理解。【参考方案3】：

在协作中运行这两个命令

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

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

!nvidia-smi

【讨论】：

以上是关于如何在 Google Colab 中获得分配的 GPU 规格的主要内容，如果未能解决你的问题，请参考以下文章