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

What about the hardware configuration? #5

Open
qinxiaoyu222 opened this issue Jun 1, 2018 · 7 comments
Open

What about the hardware configuration? #5

qinxiaoyu222 opened this issue Jun 1, 2018 · 7 comments

Comments

@qinxiaoyu222
Copy link

Hardware operating environment is a number of gpu or server?

@zouchuhang
Copy link
Owner

@qinxiaoyu222 We train & test our network on a single NVIDIA Titan X GPU. You shall be able to run it on a GPU with 12GB memory.

@qinxiaoyu222
Copy link
Author

Thank you very much. My GPU is NVIDIA GtX1080 with 8GBmemory.

@zouchuhang
Copy link
Owner

zouchuhang commented Jun 1, 2018

@qinxiaoyu222 I think you encounter the "out of memory" issue, our model use nearly all the 12GB memory for better performance. You can try down sampling the resolution of the input image by a factor of 2 and change the related network structure to fit this resolution. In this way you shall be able to run the experiments. Alternatively you can try it on a GPU with 12GB memory.

@qinxiaoyu222
Copy link
Author

@zouchuhang Thank you very much, I try it.

@zoubochao
Copy link

Mine is 1080Ti with 11GB memory, and I still get the "out the memory" error, is there anyway to work around besides down sampling the resolution? thx

@zouchuhang
Copy link
Owner

@zoubochao I checked "driver_pano_joint.lua" and it needed around 10.969GB to run, nearly touch to your 11GB memory limitation. If you don't want to down sample the resolution, it's better to locally get a Titan xp GPU with 12GB memory, or try running the network on paid public servers like AWS.

@zouchuhang
Copy link
Owner

You can try down sampling the resolution of the input image by a factor of 2 and change the related network structure to fit this resolution. In this way you shall be able to run the experiments. Alternatively you can try it on a GPU with 12GB memory.

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

3 participants