Skip to content

Commit

Permalink
Add api_difference doc
Browse files Browse the repository at this point in the history
  • Loading branch information
co63oc committed Jul 5, 2023
1 parent 9818a5b commit b89f5b5
Showing 1 changed file with 40 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,40 @@
## [参数不一致]torch.cuda.device

### [torch.cuda.device](https://pytorch.org/docs/1.13/generated/torch.cuda.device.html#torch.cuda.device)

```python
torch.cuda.device(device)
```

### [paddle.CUDAPlace](https://www.paddlepaddle.org.cn/documentation/docs/zh/develop/api/paddle/CUDAPlace_cn.html)

```python
paddle.CUDAPlace(id)
```

其中 Pytorch 与 Paddle 的参数支持类型不一致,具体如下:

### 参数映射

| PyTorch | PaddlePaddle | 备注 |
| ------- | ------------ | -------------------------------------------------------------------------------- |
| device | id | GPU 的设备 ID, Pytorch 支持 torch.device 和 int,Paddle 支持 int,需要进行转写。 |

### 转写示例

#### device: 获取 device 参数,对其取 device.index 值

```python
# Pytorch 写法
torch.cuda.device(torch.device('cuda').index)

# Paddle 写法
paddle.CUDAPlace(0)

# 增加 index
# Pytorch 写法
torch.cuda.device(torch.device('cuda', index=1).index)

# Paddle 写法
paddle.CUDAPlace(1)
```

0 comments on commit b89f5b5

Please sign in to comment.