Is your code running fine now using a single GPU?
If so, could you post the batch size and input shape you are currently using?
I’ve just got feedback that the 3D dataset in this course eats up nearly all the memory on 4 V100s.
Could thus an OOM error still be the cause of this issue for your use case?