如何找到 Google COLAB GPU 的核心数？

Question

我想知道 Colab 分配的 GPU 中的核心数。我使用了 !nvidia-smi -q，但它显示了所有其他细节，但没有显示 GPU 的核心数。

我已经对此进行了研究，但没有找到。 对此有任何帮助！！

Answer 1

最简单的方法是 运行 !nvidia-smi 命令获取所有 GPU 信息。在那之后你将获得的卡上似乎有 2 个可能的选项 - K80 或 T4，K80 有 4992 个 CUDA 内核，而 T4 有 2560 个 CUDA 内核（使用 Google 找到的）。

如果您仍然需要脚本来查明核心数量，您可以在 and 中找到这些脚本。我仍然建议使用更简单的方法来查找特定模型的 CUDA 内核。

在旁注中， 用户只能获得 5% 的 GPU 内存供他们访问。我没有遇到过这个问题，但您仍然应该检查 !nvidia-smi 命令中显示的 Volatile GPU Util（0% 是当没有 运行ning 时应该预期的结果）。