1; 2; 3; 4; 5. 从gpu转到cpu上. a = torch.arange(10).cuda() ''' device(type='cuda', index=0) 调用cuda()方法后Tensor在gpu上计算''' a = a.cpu() ...
確定！ 回上一頁
