linux下使用pytorch框架出现cuda run out of memory问题

1.使用pytorch，数据量是图像224x224，总共4w张，框架使用的是VGG，出现cuda memory问题

上图是gpu使用的情况，运行时使用的batch_size为32

显存问题怎么解决呢？

求大神指点指点，实在不知道怎么解决了。。。

检查一下是不是一次性把所有的图片都加载进去了