标签: python linux process cuda
在python中,Linux下是否有一种方法可以在每次新进程启动时获取信号?特别是在Nvidia GPU上具有计算上下文的进程何时开始?还是我需要使用一个调用ps或nvidia-smi的循环?
答案 0 :(得分:1)
NVIDIA维护一个API,用于从名为NVML(NVIDIA Management Library)的用户代码中查询CUDA生态系统的状态。 我不知道该API的任何Python绑定,但是通过您选择的方法(例如this或this)使用它并不难。您将需要轮询,它不支持信令。我认为nvidia-smi在内部使用NVML。
我不相信有任何一种Linux用户空间API可以通知新进程创建,启用CUDA或其他。