gpustat

gpustat

就比 nvidia-smi低一點？

注意：這僅適用於 NVIDIA 圖形設備，目前不支援 AMD。歡迎貢獻！

自我推銷： gpustat的 Web 介面可用（alpha 版）！查看 gpustat-web。

快速安裝

從 PyPI 安裝：

 pip install gpustat

如果您沒有 root (sudo) 權限，請嘗試在使用者命名空間上安裝gpustat ： pip install --user gpustat 。

透過 pip 安裝最新版本（master 分支）：

 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 是 NVIDIA 的 NVML 官方 python 綁定。

• 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 : 強制彩色輸出（即使 stdout 不是 tty）
• --no-color ：抑制彩色輸出
• -u , --show-user : 顯示進程擁有者的使用者名稱
• -c , --show-cmd : 顯示進程名稱
• -f , --show-full-cmd : 顯示正在運行的進程的完整命令和 cpu 統計信息
• -p , --show-pid : 顯示進程的PID
• -F , --show-fan : 顯示 GPU 風扇速度
• -e , --show-codec ：顯示編碼器和/或解碼器利用率
• -P , --show-power ：顯示 GPU 功耗和/或限制（ draw或draw,limit ）
• -a , --show-all : 顯示上面所有的 GPU 屬性
• --id ：僅使用指定索引定位和查詢特定 GPU（例如--id 0,1,2 ）
• --no-processes : 不顯示進程資訊（使用者、記憶體） (#133)
• --watch 、 -i 、 --interval ：如果給定，則以監視模式運作（相當於watch gpustat ）。表示更新之間的間隔。
• --json : JSON 輸出 (#10)
• --print-completion (bash|zsh|tcsh) : 列印 shell 完成腳本。使用方法請參閱#131。

尖端

• 如果出現問題，請嘗試gpustat --debug 。
• 若要定期監視，請嘗試gpustat --watch或gpustat -i (#41)。
  • 對於舊版本，可以使用watch --color -n1.0 gpustat --color 。
• 執行nvidia-smi daemon （需要 root 權限）將使查詢 GPU 的速度更快並使用更少的 CPU (#54)。
• gpustat （和nvidia-smi ）顯示的 GPU ID（索引）是 PCI 總線 ID，而 CUDA 預設使用不同的排序（分配最快的 GPU 和最低 ID）。因此，為了確保 CUDA 和gpustat使用相同的 GPU 索引，請將CUDA_DEVICE_ORDER環境變數配置為PCI_BUS_ID （在為 CUDA 程式設定CUDA_VISIBLE_DEVICES之前）： export CUDA_DEVICE_ORDER=PCI_BLE_DEVICES 之前）： export CUDA_DEVICE_ORDER=PCI_BUS_ID 。

預設顯示

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

• [0] ：GPU索引（從0開始）作為PCI_BUS_ID
• GeForce GTX Titan X ：GPU 名稱
• 77°C ：GPU 溫度（攝氏）
• 96 % ：GPU 使用率
• 11848 / 12287 MB ：GPU 記憶體使用量（已使用/總計）
• python/... ：在 GPU、擁有者/cmdline/PID 上執行進程（及其 GPU 記憶體使用情況）

變更日誌

請參閱 CHANGELOG.md

執照

麻省理工學院許可證