映射文档 No.120 #5966
Merged
映射文档 No.120 #5966
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
|
感谢你贡献飞桨文档，文档预览构建中，Docs-New 跑完后即可预览，预览链接：http://preview-pr-5966.paddle-docs-preview.paddlepaddle.org.cn/documentation/docs/zh/api/index_cn.html |
ROckDog22
approved these changes
Jun 27, 2023
ROckDog22
approved these changes
Jun 28, 2023
zhwesky2010
reviewed
Jul 5, 2023
| paddle.CUDAPlace(id) | ||
| ``` | ||
|
|
||
| 其中 Pytorch 与 Paddle 的参数类型不一致，具体如下： |
|
|
||
| | PyTorch | PaddlePaddle | 备注 | | ||
| | ------- | ------------ | ------------------------------------------------------- | | ||
| | device | id | GPU 的设备 ID, PyTorch 比 Paddle 多 torch.device 类型。 | |
There was a problem hiding this comment.
Pytorch支持torch.device和int，而Paddle仅支持int。需要进行转写。
然后下面针对 torch输入torch.device时的情况，写一下 转写示例
zhwesky2010
reviewed
Jul 5, 2023
|
|
||
| # 增加 index | ||
| # Pytorch 写法 | ||
| torch.cuda.device(torch.device('cuda', index=1)) |
There was a problem hiding this comment.
这个转写方式是：获取device参数，对其取device.index值
改成那种吧
zhwesky2010
reviewed
Jul 5, 2023
|
|
||
| ```python | ||
| # Pytorch 写法 | ||
| torch.cuda.device(torch.device('cuda').index) |
There was a problem hiding this comment.
这样吧
torch.cuda.device(torch.device('cuda'))
paddle.CUDAPlace(0)
torch.cuda.device(torch.device('cuda', index=index))
paddle.CUDAPlace(index)
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
PaddlePaddle/PaConvert#112
api是 torch.cuda.device