python怎么在跑深度学习时显示GPU占用率

可以使用nvidia-smi命令来查看GPU的占用率。

在Python中，可以使用以下代码来获取GPU的占用率：

import subprocess

def get_gpu_usage():
    result = subprocess.run(['nvidia-smi', '--query-gpu=utilization.gpu', '--format=csv,noheader'], stdout=subprocess.PIPE)
    gpu_usage = [int(x) for x in result.stdout.decode().strip().split('\n')]
    return gpu_usage

这个函数会返回一个列表，其中每个元素是一个GPU的占用率（以百分比表示）。如果有多个GPU，则列表中会有多个元素。

你可以在代码中的任何地方调用该函数来获取GPU的实时占用率。例如，你可以在训练模型的循环中调用它来监测GPU的使用情况：

for epoch in range(num_epochs):
    for batch_idx, (data, target) in enumerate(train_loader):
        # 训练模型
        # ...
        
        # 每隔一定时间检查GPU占用率
        if batch_idx % check_interval == 0:
            gpu_usage = get_gpu_usage()
            print(f'GPU usage: {gpu_usage}')