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.

Sign up for GitHub

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

Jump to bottom

GPU训练 #58

Open
wxue1201 opened this issue May 6, 2022 · 1 comment
Open

GPU训练 #58

wxue1201 opened this issue May 6, 2022 · 1 comment

Comments

@wxue1201
Copy link

wxue1201 commented May 6, 2022 •
edited
Loading

请问大佬，能否使用GPU训练模型？我的torch.cuda.is_available()为True,代码指定GPU这一行 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")即指定GPU，但当训练时，nvidia-smi查看GPU利用率是为0，请问博主是我操作上忘记某个开启GPU的开关了吗？还请大佬提示
@furkansezer0
Copy link

@wxue1201 did you solve the problem?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@wxue1201 @furkansezer0