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

8 浏览
0 Comments

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

我正在使用Google Colab进行深度学习,并且我知道他们会随机分配GPU给用户。我想知道在任何给定的会话中我被分配了哪个GPU。在Google Colab笔记本中有办法做到这一点吗?请注意,我正在使用Tensorflow。

0