nvsmi

是nvidia-smi的一个(n用户-)友好的包装器。

它可以用来根据资源使用情况过滤GPU（例如，在多GPU系统中选择利用率最低的GPU）。

用法

命令行界面

nvsmi --help
nvsmi ls --help
nvsmi ps --help

作为库使用

import nvsmi

nvsmi.get_gpus()
nvsmi.get_available_gpus()
nvsmi.get_gpu_processes()

先决条件

• 一个NVIDIA GPU
• nvidia-smi
• Python 2.7或3.6+

安装

pipx

推荐的安装方法是pipx。具体来说，您可以使用以下命令为您自己的用户安装nvsmi

pipx install nvsmi

上述命令将在~/.local/pipx/venvs/nvsmi中创建一个虚拟环境，并将nvsmi可执行文件添加到~/.local/bin中。

pip

或者您可以使用老式的pip，但这比pipx更脆弱

pip install --user nvsmi