how to train with multiple gpus

Jamie-Cheung / ArtBank

ArtBank: Artistic Style Transfer with Pre-trained Diffusion Model and Implicit Style Prompt Bank (AAAI2024)

31 stars 2 forks source link

Open zhuangtu1203 opened 8 months ago

zhuangtu1203 commented 8 months ago

My graphics card is showing that out of memory, how can I put this on the multi-GPU for training

Jamie-Cheung commented 8 months ago

My graphics card is showing that out of memory, how can I put this on the multi-GPU for training

Each graphics card needs at least 20 GB memory for training. Multi-card training also requires each graphics card to 20 GB memory.