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