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

why not use gpu #382

Open
nana1223 opened this issue Dec 14, 2023 · 2 comments
Open

why not use gpu #382

nana1223 opened this issue Dec 14, 2023 · 2 comments

Comments

@nana1223
Copy link

您好，想请问下，我在跑需求预测任务用DMVSTNet模型跑TAXIBJ数据集，但是却没有用到gpu，一直在用cpu跑，查看了代码中的配置确实是在使用gpu的呀，想请问下这种情况怎么去修改它命中gpu呢？（已确认本机有gpu且pytorch也是下的gpu版本的）感谢回复！
@Blore-lzn
Copy link

先看下torch.cuda.is_available()的结果是True吗

@nana1223
Copy link
Author

先看下torch.cuda.is_available()的结果是True吗

是true的

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
@nana1223 @Blore-lzn