gpu - 在 Google Colaboratory 中，有没有办法表明偏好或需要特定的 gpu 模型？

Question

我正在尝试为某些任务测量 Google Colaboratory 的运行时扩展和限制，这可能都很大程度上取决于后端分配的 gpu 模型。目前在 notebook 中，Runtime，Change runtime type，Hardware Accelerator 带来了none、gpu、tpu 选项。那里没有能力表明对特定 gpu 模型的偏好或要求。是否有其他方法可以表明对特定 gpu 模型的偏好或要求，或者对会话分配给 VM 的 gpu 模型不感兴趣？拥有这样的能力会很棒。目前，我只是不断尝试新的会话，直到我得到我正在寻找的 gpu 模型来执行任务。这对于 Google 基础架构和用户来说都是乏味且低效的。

标签： gpubenchmarkinggoogle-colaboratory