search

dribnet / pixray

neural image generation
Other
402 stars 53 forks source link

Multi GPU support? #14

Open daniel-chabanenko opened 2 years ago

daniel-chabanenko commented 2 years ago

Is there a possibility to run this on multi-gpu setup? Right now it seems like utilising only GPU 0 for me

Thanks a lot

dribnet commented 2 years ago

This isn't currently possible. In the future this might be interesting to consider - perhaps putting different CLIP models on different GPUs, etc.

(FWIW: what I find I do instead currently with multiple GPUs is run different variations as separate processes using CUDA_VISIBLE_DEVICES)