nvidia smi的(用户友好的)包装
nvsmi的Python项目详细描述
NVSMI
对nvidia-smi
的(用户友好的)包装。
用法
cli
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
和
在~/.local/bin
中添加nvsmi
可执行文件。
PIP
或者你可以使用好的旧的pip
但是这比pipx
更脆弱:
pip install --user nvsmi