[参数不一致]torch.cuda.device¶
torch.cuda.device¶
torch.cuda.device(device)
参数映射¶
PyTorch | PaddlePaddle | 备注 |
---|---|---|
device | id | GPU 的设备 ID, Pytorch 支持 torch.device 和 int,Paddle 支持 int,需要转写。 |
torch.cuda.device(device)
PyTorch | PaddlePaddle | 备注 |
---|---|---|
device | id | GPU 的设备 ID, Pytorch 支持 torch.device 和 int,Paddle 支持 int,需要转写。 |