要查看gpu占用,这里介绍两个方法,一个是官方的nvidia-smi,装完显卡驱动就能使用了;另一个是gpustat,更方便于查看gpu使用情况。
1 nvidia-smi
在终端输入命令nvidia-smi
,可以查看显卡使用情况:

但是,这样只能查看某一时刻的显卡使用情况。为了实时显示显卡的使用情况,可以添加参数。在终端输入命令:
watch -n 5 nvidia-smi
这样系统就会每隔5秒刷新一次使用情况。这里的5
可以依据需要进行修改。
如上图所示,在下面的虚线框里显示了占用GPU的进程信息。每个进程有一个唯一的PID,如果想关闭掉某个进程,可以使用命令: