无法导入 nvidia_smi

Question

我倾向于在 python 代码运行期间收集 GPU 状态。我需要import nvidia_smi在我的代码中执行此操作。但即使通过安装它pip install nvidia_smi遇到此错误： No module named 'nvidia_smi'

任何想法？

Answer 1

我在这里找到了答案pip install nvidia-ml-py3：

和：

import nvidia_smi

nvidia_smi.nvmlInit()
handle = nvidia_smi.nvmlDeviceGetHandleByIndex(0)