[PyTorch] Release GPU / CPU Memory After Delete Model
Last Updated on 2024-01-15 by Clay
Problem
Yesterday, I developed a model merging program. This time I have no enough gpu memory to merge the models in only one time, so I tried to merge layer by layer. I found the memory of GPU is easily to release but CPU didn't.
Read More »[PyTorch] Release GPU / CPU Memory After Delete Model