Как получить выделенный GPU spe c в Google Colab - PullRequest
0 голосов
/ 19 февраля 2020

Я использую Google Colab для глубокого изучения и знаю, что они случайным образом распределяют графические процессоры пользователям. Я хотел бы видеть, какой графический процессор мне выделен в любой сессии. Есть ли способ сделать это в записных книжках Google Colab?

Обратите внимание, что я использую Tensorflow, если это поможет.

Ответы [ 2 ]

4 голосов
/ 19 февраля 2020

Поскольку вы можете запустить команду bash в colab, просто запустите !nvidia-smi: enter image description here

1 голос
/ 20 февраля 2020

Это облегчает чтение

!nvidia-smi -L
...