[ 输入参数类型不一致 ]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。