我正在尝试使用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
答案 0 :(得分:0)
您可以通过以下方式释放现有内存
nvidia-smi
记录该过程并使用
终止sudo kill -9 <pid>