python - 从 Torch/Tensorflow 设备中查找 PID 号

Question

我正在从三个不同的 Jupyter Notebooks（我们称它们为 JN1、JN2 和 JN3）（python 3.6）运行三个 PyTorch 模型。所有模型都在同一个 GPU (Titan Xp) 上运行，这是一个共享资源，托管在不同的服务器上。

我完成了其中一个模型/笔记本（JN1），并想单独关闭与此模型对应的 PID 进程。nvidia-smi 给了我一个 PID 进程及其编号的列表 - 但是，我不确定哪个是对应于 JN1 的 PID 编号。

有没有办法直接从笔记本或手电筒型号中找到该型号的PID号？

标签： pythontensorflowgpupidtorch