电脑GPU/CPU资源查看及使用

电脑GPU/CPU资源查看及使用

import torch
'''
1. 服务器GPU状态查询

    1） lspci | grep -i nvidia 可以查询所有nvidia显卡

    2） lspci -v -s [显卡编号] 可以查看显卡具体属性

    3） nvidia-smi 可以查看显卡的显存利用率

2. torch.cuda主要函数

    1) 查看是否有可用GPU、可用GPU数量： torch.cuda.is_available(), torch.cuda.device_count()

    2) 查看当前使用的GPU序号：torch.cuda.current_device()

    3) 查看指定GPU的容量、名称：

        torch.cuda.get_device_capability(device),  torch.cuda.get_device_name(device)

    4) 清空程序占用的GPU资源： torch.cuda.empty_cache()

    5) 为GPU设置随机种子：torch.cuda.manual_seed(seed), torch.cuda.manual_seed_all(seed)

'''

def select_device(id):
    force_cpu = False
    if id == -1: force_cpu = True
    cuda = False if force_cpu else torch.cuda.is_available()    #如果电脑不存在GPU，即使id不为零，cuda也是False
    device = torch.device('cuda:{}'.format(id) if cu