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

关于使用CUDA的问题 #48

Closed
Kepler0330 opened this issue Apr 24, 2021 · 2 comments
Closed

关于使用CUDA的问题 #48

Kepler0330 opened this issue Apr 24, 2021 · 2 comments

Comments

@Kepler0330
Copy link

作者您好，我使用2080ti且把argument中的cuda设置为True了，使用torch.cuda.is_available()测试返回也是True，但是每个算法每一张图每一次run（八次中的一次）还是要跑七八个小时，想问你们当初也是这个速度吗（不知道是我的问题还是大家都这样）
@Kepler0330
Copy link
Author

⬆️并且gpu利用率也不足10%

@starry-sky6688
Copy link
Owner

正常，你可以通过控制n_steps来决定训练多久，一般200万个timestep需要一天左右才能跑完；

GPU利用率低是因为batch_size=32，你调大一些就好了

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
@starry-sky6688 @Kepler0330