-
Notifications
You must be signed in to change notification settings - Fork 55
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
【Hackathon 5th No.45】convert torch.cuda.get_device_name and torch.cuda.device -part #311
Conversation
Thanks for your contribution! |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
由于CI只支持CPU,GPU的API,需要把本地测试结果截图贴上来
@@ -4400,6 +4400,16 @@ | |||
"device": "device" | |||
} | |||
}, | |||
"torch.cuda.get_device_name": { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
这两个API返回的结果是一致的吗?看起来properties包含额外的信息
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
重新修改好了
看起来并没有完成torch.cuda.device的内容? |
是的,还没有完成 |
result = None | ||
if torch.cuda.is_available(): | ||
current_device = torch.cuda.current_device() | ||
result = torch.cuda.get_device_name(current_device) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
单测本地运行截图有吗
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
PR Docs
PaddlePaddle/docs#6217
PR APIs
torch.cuda.get_device_name的文档 get_device_name