lei_qi的博客

1、输入命令fuser -v /dev/nvidia*查看显卡占用2、使用命令kill -9 [PID]删除占用进程的PID（即上图中的PID一列

指定GPU 进行深度学习训练# 方法1：torch.cuda.set_device(1)# 方法2：device = torch.device("cuda:1")# 方法3：# 推荐使用os.environ["CUDA_VISIBLE_DEVICES"] = '1'多GPU并行代码使用四卡进行并行训练 多GPU并行 os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3" if torch.cuda.device_count()

每隔1秒动态刷新GPU使用情况watch -n 1 nvidia-smi