pytorch 直接申请gpu空间 Posted on 2018-08-Sun Views: Views: 1mat = torch.zeros((51, 48), dtype=torch.float, device=device)