运行时错误:CUDA内存不足。试图分配

2024-04-26 13:56:39 发布

您现在位置:Python中文网/ 问答频道 /正文

我正试图用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


Tags: of代码gpu错误笔记本jupyterpytorch批量