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

显卡问题 #3

Closed
epiphany1111 opened this issue Feb 23, 2023 · 2 comments
Closed

显卡问题 #3

epiphany1111 opened this issue Feb 23, 2023 · 2 comments

Comments

@epiphany1111
Copy link

您好，非常感谢能够看到这样优秀的工作，有一点想要请教，我把项目代码放在3090显存为24G的单卡上运行，出现cuda out of memory的错误，故想向作者询问一下在训练时使用的显卡，以及代码需要多大的显存才能运行。
@JinghaoZ
Copy link
Collaborator

你好，感谢你对我们工作的关注！
我们使用的是32G的V100训练；此外，我们在扩展文章的代码 https://github.com/CRIPAC-DIG/MICRO 中增加了稀疏化构图操作，可以在24G的单卡上运行。

@epiphany1111
Copy link
Author

非常感谢您的解答

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
@JinghaoZ @epiphany1111