RuntimeError:CUDA内存不足。试图分配

时间:2020-06-17 04:35:56

标签: python error-handling pytorch runtime

我正在尝试使用jupyter笔记本运行pytorch代码,但出现此错误

RuntimeError: CUDA out of memory. Tried to allocate 20.00 MiB (GPU 0; 2.00 GiB total capacity; 1.23 GiB already allocated; 18.83 MiB free; 1.25 GiB reserved in total by PyTorch)

我已经找到答案了。最重要的是,只是减少了批量。 我尝试将批量大小从20减少到10到2和1。现在仍然无法运行代码

我的GPU是Quadro p620

1 个答案:

答案 0 :(得分:0)

您可以通过以下方式释放现有内存

nvidia-smi

记录该过程并使用

终止
sudo kill -9 <pid>