pytorch解决cuda显存问题的方法

使用jupyter notebook
之前前面几百次迭代都没问题，但显存越来越大，最后炸了，这还亏着是4090
我觉得是cuda自动清理显存机制有问题，一直不自动清理
方法尝试：每轮迭代之后使用一次：

torch.cuda.empty_cache()

此外，jupyter notebook结束进程后不会自动清理显存，还需

import os
 
pid = os.getpid()
!kill -9 $pid