pytorch深度学习单卡训练和多卡训练

单机单卡训练模式

 # 设置GPU参数,是否使用GPU,使用那块GPU
    if config.use_gpu and torch.cuda.is_available():
        device=torch.device('cuda',config.gpu_id)
    else:
        device=torch.device('cpu')
    # 检查一下GPU是否可以使用
    print('GPU是否可用:'+str(torch.cuda.is_available()))
 # 设置GPU参数,是否使用GPU,使用那块