[ 输入参数类型不一致 ]torch.cuda.set_device¶

torch.cuda.set_device ¶

torch.cuda.set_device(device)

paddle.device.set_device(device)

功能一致，参数类型不一致，具体如下：

PyTorch	PaddlePaddle	备注
device	device	PyTorch 支持 torch.device 或 int。PaddlePaddle 支持 str。