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.

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

请问对于fast testing中的测试大概需要多大的显存? #88

Open
Rhhhhhhh opened this issue Aug 15, 2024 · 2 comments
Open

Comments

@Rhhhhhhh
Copy link

我使用的是一张832*480的png图片进行Real-world image super-resolution的测试,使用单卡4090和A6000分别进行测试,参数为scale 4 version v3,一直提示爆显存,大概需要多少显存才够呢?还是说至少需要多少显存?

@zsyOAOA
Copy link
Owner

zsyOAOA commented Aug 15, 2024

  1. 可以调整chop_size(https://github.com/zsyOAOA/ResShift/blob/journal/inference_resshift.py#L53)的大小来调整占用的显存,默认是512.
  2. 检查一下是否安装了xformers, 可以节省很多显存。 @Rhhhhhhh

@Rhhhhhhh
Copy link
Author

十分感谢!调整了chop_size之后成功了!
xformers应该是有安装的,是按照readme中的requirements.txt安装的包,应该是有包含的。具体代码细节我再研究研究

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants