如何在 Colab 中获得 V100 GPU？

Question

我买了 Colab Pro，而我大部分时间只能申请 P100。 V100 的运行速度几乎是 P100 的 2 倍。 如何手动获取 V100？

Answer 1

我也一直在使用 Colab Pro，据我所知，这些资源是根据 Google 的可用性分配的。 我大部分时间都在使用Tesla P100-PCIE-16GB ，但随机分配给我一个Tesla V100-SXM2-16GB 。

顺便说一句，要打印您的设备名称，您可以在 Pytorch 中使用此命令：

import torch
 
torch.cuda.get_device_name(device=None)

Answer 2

我开始使用 Colab pro 大约三个月了，从那以后，我什至没有一次得到 V100，而且大部分时间，我得到了 P100，有时得到了 T4。

要获得您在 Colab 中使用的 GPU，最好的方法是使用以下命令：

!nvidia-smi