英伟达显卡查看占用情况
1. 查看显存占用情况:
运行该命令nvidia-smi
显示所有显卡的占用率
相关参数介绍如下
-i, --id= Target a specific GPU or Unit.
-f, --filename= Log to a specified file, rather than to stdout.
-l, --loop= Probe until Ctrl+C at specified second interval.
-lms, --loop-ms= Probe until Ctrl+C at specified millisecond interval.
想要实时显示利用率,可以用以下命令
nvidia-smi -l 1
2 设置使用的显卡
export CUDA_VISIBLE_DEVICES= XXXX
使用该命令后,其余显卡不可见