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

torch.cuda.device

torch.cuda.device(device)

paddle.device._convert_to_place

paddle.device._convert_to_place(device)

其中 PyTorch 与 Paddle 的参数支持类型不一致，具体如下：

参数映射

PyTorch PaddlePaddle 备注
device device GPU 的设备 ID, PyTorch 支持 torch.device 和 int，Paddle 支持 str，需要转写。

转写示例

device: 特定的运行设备

# PyTorch 写法
torch.cuda.device('cuda:0')

# Paddle 写法
paddle.device._convert_to_place('gpu:0')

# PyTorch 写法
torch.cuda.device(2)

# Paddle 写法
paddle.device._convert_to_place('gpu:2')