NeuralNeighborStyleTransfer icon indicating copy to clipboard operation
NeuralNeighborStyleTransfer copied to clipboard

Multi GPU mode?

Open surfingnirvana opened this issue 2 years ago • 2 comments

Is it possible to use two GPUs?

surfingnirvana avatar Apr 18 '22 09:04 surfingnirvana

Hi, Currently it isn't, naively I'm not sure if the algorithm is amenable to being split across GPUs. I suppose you could do it in tiles, but I'm not sure how you'd handle overlap. If you have suggestions I'd be very open to them! I assume you're interested in multiple GPUs to scale it to higher resolutions?

nkolkin13 avatar Apr 23 '22 01:04 nkolkin13

Just want to say that your variant of Style Transfer really good! i'm not programmer but i keep eyes on style transfer variations. your variant most simple to install and produces very good quality. Sadly most of "game" GPU's is something around 8-10gb - that makes unable to generate 1024px image. (usually we dont need images higher rhan 1024 because AI upscale can generate 2-4k images from 1024px without loosing too much quality). p.s. - interesting that rtx 3060 have 12gb memory. but 3060ti/3070 is 8gb. I tested my rtx 3070 - max dimension of cube is 640px, bigger values return error (not enough memory).

Hi, Currently it isn't, naively I'm not sure if the algorithm is amenable to being split across GPUs. I suppose you could do it in tiles, but I'm not sure how you'd handle overlap. If you have suggestions I'd be very open to them! I assume you're interested in multiple GPUs to scale it to higher resolutions?

DogVanDog avatar Apr 29 '22 18:04 DogVanDog