在 python 中,Linux 下有没有办法在每次启动新进程时获取信号?特别是当 Nvidia GPU 上具有计算上下文的进程启动时?或者我是否需要使用调用 ps 或 nvidia-smi 的循环?
最佳答案
NVIDIA 维护一个 API,用于从名为 NVML 的用户代码中查询 CUDA 生态系统的状态。 (NVIDIA 管理库)。 我不知道该 API 有任何 Python 绑定(bind),但通过您选择的方法(例如 this 或 this )使用它不会太困难。您需要轮询,它不支持信令。我假设 nvidia-smi 在内部使用 NVML。
我不相信有任何类型的 Linux 用户空间 API 允许通知新进程创建、CUDA 启用或其他。
关于python - 当任何新进程开始时得到一个信号,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51069017/