How much GPU memory need for run python main.py?

[lood339]

Hi,
I tried to run the code and see the result in a 16 Gb GPU memory machine.
But it shows:
...
Processing quantile index 63296: 758.747045 2169.666316
Exceeded job memory limit
and stops.
Can I ask the memory requirement of the program?
Thank you.

Jimmy

[sunyiyou]

If you are running Resnet18, 16 Gb GPU memory is totally enough. It possibly exceeded your main memory.

[lood339]

I am double checking the problem as I have a very large main memory (64G). May I ask the typical machine configuration (CPU, GPU) on which the code runs well?
Thank you.

[sunyiyou]

32GB main memory and 12GB GPU memory works(tested). So I have no clue about the error reported.