解决CUDA out of memory

问题 用Pytorch跑模型时，会出现RuntimeError: CUDA out of memory 解决方法 1.真的没有内存了 可以改小batch_size 2.内存足够，不分配 ...