instruct-pix2pix icon indicating copy to clipboard operation
instruct-pix2pix copied to clipboard
verified publisher icon timothybrooks

What are the minimum GPU memory requirements for running this InstructPix2Pix training setup?

Open fwbc opened this issue 8 months ago • 4 comments

fwbc avatar May 05 '25 08:05 fwbc

I used RTX 4090 24G, but always got an error when operate multi_tensor

baihuple avatar May 14 '25 15:05 baihuple

I used RTX 4090 24G, but always got an error when operate multi_tensor

as same as you,alway Out Of Memory when I run train code .I think maybe his code has some bug

fwbc avatar May 15 '25 00:05 fwbc

32GB should be enough.

I don't know the minimum GPU memory for running But I ran it on a single GPU with 32GB of memory, but I closed it before completing an epoch since it was excessively prolonged. I finally used multiple GPUs in parallel to barely run a couple of epochs.

I hope my experience can help you.

yukiime avatar May 20 '25 20:05 yukiime

Finally，I use single L20 (48G), and got this

Image

baihuple avatar May 21 '25 02:05 baihuple