LayoutNet icon indicating copy to clipboard operation
LayoutNet copied to clipboard

What about the hardware configuration?

Open qinxiaoyu222 opened this issue 7 years ago • 7 comments

Hardware operating environment is a number of gpu or server?

qinxiaoyu222 avatar Jun 01 '18 09:06 qinxiaoyu222

@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.

zouchuhang avatar Jun 01 '18 16:06 zouchuhang

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

qinxiaoyu222 avatar Jun 01 '18 17:06 qinxiaoyu222

@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.

zouchuhang avatar Jun 01 '18 17:06 zouchuhang

@zouchuhang Thank you very much, I try it.

qinxiaoyu222 avatar Jun 02 '18 05:06 qinxiaoyu222

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

zoubochao avatar Jul 25 '18 05:07 zoubochao

@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 avatar Jul 25 '18 06:07 zouchuhang

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.

zouchuhang avatar Dec 08 '18 01:12 zouchuhang