Closed AlphaPlusTT closed 4 years ago
i use single gpu and set the samples_per_gpu as 1, but the model occupy almost 12G gpu memory!
you can use 'lsof /dev/nvidia*' to check the running process on the GPU, I think it might because there're dead process left in your GPU memory.
i use single gpu and set the samples_per_gpu as 1, but the model occupy almost 12G gpu memory!