search

manuelruder / artistic-videos

Torch implementation for the paper "Artistic style transfer for videos"
Other
1.75k stars 252 forks source link

Multi-GPU update? #36

Open bododge opened 7 years ago

bododge commented 7 years ago

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.

NameRX commented 7 years ago

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.

NameRX commented 7 years ago

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

filmo commented 7 years ago

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.

HandsomeDevilv112 commented 7 years ago

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.