自动检查Google上可用的GPU

Question

是否有一种自动的方法来检查哪个GPU目前在Google (Pro)上可用。

假设我想使用特斯拉P100而不是特斯拉T4来训练我的模型，有没有办法定期检查Colab中的python脚本，P100是否可用？

我已经尝试过周期性地消除内核，但是在关闭之后它不会自动重新启动：

import os

def restart_runtime():
   os.kill(os.getpid(), 9)

谢谢

Answer 1

没有办法检查什么GPU可用。增加一行：

!nvidia-smi

然后继续断开和重新连接运行时，直到得到所需的GPU为止。