Multi-GPU update?
bododge opened this issue · 4 comments
I am wondering if it would be an easy fix to incorporate the new multi-gpu functionality jcjohnson has built into neural-style into your video script? It would be nice to have the option of using both gpu's to create larger images.
I think too, that this will be a very useful option for this tool. GTX 1070 with 8Gb VRAM can process only ~1 mpix images, for fullHD it is not enough memory, even on Titan X 12GB.
I suppose it is not a big problem to incorporate multi-gpu support, probably just need to add function setup_multi_gpu
from neural_style.lua and setup network by net = setup_multi_gpu
.
upd.: not so easy, as i thought... need authors help to add this functionality
Interesting. I just had the same question.
Is there something fundamental about this implementation that requires the model be on a single GPU or is it just a matter of my spending time to implement it. I see that @NameRX may have already given it a shot.
@NameRX what did you find that was not so easy? (Coding, or a fundamental problem the prevent implementation.) I'm interested in giving it a shot, provided that it's actually a doable thing but would appreciate your insight as to what stopped you.
Did anybody ever figure out how to make this work/find a work around? I just installed a second card and would really like to use it for processing videos.