目录 1. 问题所示 2. 原理分析 3. 解决方法 3.1 调batch_size大小 3.2 不考虑梯度 3.3 删除无用变量 3.4 kill进程(暴力放弃) 3.5 其他方法 1. 问题所示 运行这段程序的时候出现如下错误: RuntimeError: CUDA out of memory. Tried to allocate 1.38 GiB (GPU 0; 7.80 GiB total capacity; 5.94 GiB already allocated; 230.25 MiB free; 6.41 GiB reserved in total by PyTorch) 截图如下所示: