198808xc / Pangu-Weather

An official implementation of Pangu-Weather
1.1k stars 202 forks source link

3060, 12g graphics memory, indicating insufficient memory during inference #63

Open Lesetoiles-lwj opened 4 months ago

Lesetoiles-lwj commented 4 months ago

Hello, during the process of using the above configuration for GPU inference, I encountered a problem of memory explosion, which is very different from my theoretical calculation of memory usage. How is the memory usage calculated in the forward inference process, or is there any other problem that caused the memory explosion?