我有一个在不同 GPU 上运行的进程的 PID 列表。我想根据每个进程的 PID 获取其使用的 GPU 内存。nvidia-smi产生我想要的信息;但是,我不知道如何 grep 它,因为输出很复杂。我已经在寻找如何做到这一点,但我还没有找到任何直接的答案。
nvidia-smi
bash cuda nvidia-smi
bash ×1
cuda ×1
nvidia-smi ×1