search

nerdyrodent / VQGAN-CLIP

Just playing with getting VQGAN+CLIP running locally, rather than having to use colab.
Other
2.61k stars 427 forks source link

Multi GPU ? #24

Open hoemon opened 3 years ago

hoemon commented 3 years ago

Could not make this run on multi GPUs. Would love some help!

rlallen-nps commented 3 years ago

I'd like to second this!

bhughesshelton commented 3 years ago

Same issue here.

arkseal commented 3 years ago

This could be useful, https://horovod.ai/ I can try to write something up

plaeground commented 3 years ago

i will donate to whoever gets this working with multiple GPUs

SA22C commented 2 years ago

i will donate to whoever gets this working with multiple GPUs

Same here, happy to donate. This needs to happen.

m-ali-awan commented 2 years ago

Hi all, hope you are fine. I am a bit confused, that how multi-gpu is going to help us. As we are not having a generic dataset, we cannot split the dataset on multiple GPUs, and also here the z space is continuously updating. So, how Multi-GPU is going to help really? Sorry, if my query sounds stupid :), but I am working on optimizing this pipeline to its best, and that's what I ended up on multi-gpu approach....

Thanks.....

joaodafonseca commented 2 years ago

any updates on this?