`gpustat`

Просто меньше , чем у нвидиа-сми?

Снимок экрана: gpustat -cp

ПРИМЕЧАНИЕ. Это работает только с графическими устройствами NVIDIA, на данный момент поддержка AMD отсутствует. Вклады приветствуются!

Самореклама: доступен веб-интерфейс gpustat (в альфа-версии)! Проверьте gpustat-web.

Быстрая установка

Установить из PyPI:

 pip install gpustat

Если у вас нет привилегий root (sudo), попробуйте установить gpustat в пространстве имен пользователя: pip install --user gpustat .

Чтобы установить последнюю версию (основную ветку) через pip:

 pip install git+https://github.com/wookayin/gpustat.git@master

Требования к драйверу NVIDIA и `pynvml`

Важный

НЕ НУЖНО: pip install pynvml и не включать pynvml в качестве зависимости в ваш проект Python. Это не сработает.

Вместо этого: pip install nvidia-ml-py . nvidia-ml-py — это официальная привязка Python для NVML от NVIDIA.

gpustat 1.2+: требуется nvidia-ml-py >= 12.535.108 (#161)
gpustat 1.0+: требуется драйвер NVIDIA версии 450.00 или выше и nvidia-ml-py >= 11.450.129 .
Если ваш драйвер NVIDIA слишком старый, вы можете использовать более старые версии gpustat ( pip install gpustat<1.0 ). См. № 107 для получения более подробной информации.

Требования Python

gpustat<1.0: совместим с Python 2.7 и >=3.4.
gpustat 1.0: Python >= 3.4
gpustat 1.1: Python >= 3.6

Использование

$ gpustat

Параметры (более подробную информацию см. gpustat --help ):

--color : Принудительно цветной вывод (даже если стандартный вывод не является tty)
--no-color : Подавить цветной вывод
-u , --show-user : показать имя пользователя владельца процесса.
-c , --show-cmd : показать имя процесса
-f , --show-full-cmd : Отобразить полную статистику команды и процессора запущенного процесса.
-p , --show-pid : показать PID процесса
-F , --show-fan : Отобразить скорость вращения вентилятора графического процессора.
-e , --show-codec : Отобразить использование кодера и/или декодера.
-P , --show-power : Отобразить использование и/или ограничение мощности графического процессора ( draw или draw,limit )
-a , --show-all : показать все свойства графического процессора, указанные выше.
--id : нацеливать и запрашивать определенные графические процессоры только с указанными индексами (например --id 0,1,2 ).
--no-processes : не отображать информацию о процессе (пользователь, память) (#133)
--watch , -i , --interval : запуск в режиме просмотра (эквивалент watch gpustat ), если он задан. Обозначает интервал между обновлениями.
--json : вывод JSON (#10)
--print-completion (bash|zsh|tcsh) : Распечатать сценарий завершения оболочки. См. № 131 для использования.

Советы

Попробуйте gpustat --debug если что-то пойдет не так.
Для периодического просмотра попробуйте gpustat --watch или gpustat -i (#41).
- Для более старых версий можно использовать watch --color -n1.0 gpustat --color .
Запуск nvidia-smi daemon (требуются права root) значительно ускорит запросы к графическим процессорам и задействует меньше ресурсов ЦП (#54).
Идентификатор графического процессора (индекс), отображаемый gpustat (и nvidia-smi ), представляет собой идентификатор шины PCI, тогда как CUDA по умолчанию использует другой порядок (назначает самый быстрый графический процессор с наименьшим идентификатором). Поэтому, чтобы гарантировать, что CUDA и gpustat используют один и тот же индекс графического процессора , настройте переменную среды CUDA_DEVICE_ORDER на PCI_BUS_ID (перед установкой CUDA_VISIBLE_DEVICES для вашей программы CUDA): export CUDA_DEVICE_ORDER=PCI_BUS_ID .

Дисплей по умолчанию

 [0] GeForce GTX Titan X | 77°C,  96 % | 11848 / 12287 MB | python/52046(11821M)

[0] : индекс графического процессора (начинается с 0) как PCI_BUS_ID.
GeForce GTX Titan X : название графического процессора
77°C : температура графического процессора (по Цельсию).
96 % : загрузка графического процессора
11848 / 12287 MB : использование памяти графического процессора (используется/всего)
python/... : запуск процессов на графическом процессоре, владелец/cmdline/PID (и использование ими памяти графического процессора)