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问题 #2

Open
shylie728 opened this issue Jun 4, 2022 · 2 comments
Open

GPU问题 #2

shylie728 opened this issue Jun 4, 2022 · 2 comments

Comments

@shylie728
Copy link

您好，我问请问一下，您说使用2块32GB的GPU来进行训练，但是我使用的是4块10GB的GPU，好像会报错，然后我尝试把batchsize调小，但又会报另外的错，请问这种情况我还可以使用我的服务器进行训练吗？
@Neverland610
Copy link
Owner

2块32GB的GPU训练能得到比较好训练效果。
如果你要用4块10GB的GPU话

  1. 模型效果会比较受限。
  2. 你需要调整batch size，不仅是在训练脚本里调，还需要根据报错信息，调整一些代码的写法，因为目前代码里有一些assert是写死的，你需要调整这部分。

@lixin2002cn
Copy link

您好，我问请问一下，您说使用2块32GB的GPU来进行训练，但是我使用的是4块10GB的GPU，好像会报错，然后我尝试把batchsize调小，但又会报另外的错，请问这种情况我还可以使用我的服务器进行训练吗？

您好 ，请问您解决这个问题了吗

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
3 participants
@Neverland610 @shylie728 @lixin2002cn