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