[ 输入参数类型不一致 ]torch.cuda.get_device_properties¶
torch.cuda.get_device_properties¶
torch.cuda.get_device_properties(device)
paddle.device.cuda.get_device_properties¶
paddle.device.cuda.get_device_properties(device)
两者功能一致但参数类型不一致，具体如下：
参数映射¶
| PyTorch | PaddlePaddle | 备注 |
|---|---|---|
| device | device | 表示设备、设备 ID 和类似于 gpu:x 的设备名称。默认值为 None，需要转写。 |
转写示例¶
device: 设备¶
# PyTorch 写法
torch.cuda.get_device_properties('cuda:0')
# Paddle 写法
paddle.device.cuda.get_device_properties('gpu:0')
# PyTorch 写法
torch.cuda.get_device_properties(2)
# Paddle 写法
paddle.device.cuda.get_device_properties('gpu:2')
# PyTorch 写法
torch.cuda.get_device_properties()
# Paddle 写法
paddle.device.cuda.get_device_properties()