Skip to content
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

在check_gpu时增加对当前环境可用设备的检查 #11293

Merged
merged 1 commit into from
Dec 6, 2023

Conversation

TracebaK
Copy link
Contributor

PR 类型 PR types

Others

PR 变化内容类型 PR changes

APIs

描述 Description

在check_gpu函数中补充了对当前环境可用设备的检查。在环境中安装了cuda版本的paddlepaddle,但没有可用的gpu时,默认使用cpu加载模型。

提PR之前的检查 Check-list

  • [√] 这个 PR 是提交到dygraph分支或者是一个cherry-pick,否则请先提交到dygarph分支。
    This PR is pushed to the dygraph branch or cherry-picked from the dygraph branch. Otherwise, please push your changes to the dygraph branch.
  • [√] 这个PR清楚描述了功能,帮助评审能提升效率。This PR have fully described what it does such that reviewers can speedup.
  • [√] 这个PR已经经过本地测试。This PR can be convered by current tests or already test locally by you.

Copy link

paddle-bot bot commented Nov 23, 2023

Thanks for your contribution!

@CLAassistant
Copy link

CLAassistant commented Nov 23, 2023

CLA assistant check
All committers have signed the CLA.

@shiyutang
Copy link
Collaborator

@TracebaK 你好,请签署CLA~

@TracebaK
Copy link
Contributor Author

@shiyutang CLA已签署~

Copy link
Collaborator

@shiyutang shiyutang left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@shiyutang shiyutang merged commit 317eac1 into PaddlePaddle:dygraph Dec 6, 2023
2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants