RuntimeError:CUDA内存不足。批量小

时间:2020-04-30 14:43:03

标签: deep-learning neural-network computer-vision

我遇到了这个错误: RuntimeError:CUDA内存不足。尝试分配116.00 MiB(GPU 0;总容量7.80 GiB;已分配1.95 GiB;免费110.00 MiB; PyTorch总共保留2.00 GiB)

但是我的批处理量已经很小,所以我不知道问题出在哪里。

我正在使用Mask RCNN网络。

当我使用nvidia-smi查看我的GPU时,这就是输出。 :| NVIDIA-SMI 440.59驱动程序版本:440.59 CUDA版本:10.2 | | ------------------------------- + ----------------- ----- + ---------------------- + | GPU名称持久性-M |总线编号Disp.A |挥发性不佳。 ECC | |风扇温度性能:用法/上限|内存使用| GPU实用计算M。|

| 0 GeForce RTX 2070关闭| 00000000:01:00.0关闭| N / A | | 20%35C P8 8W / 175W | 1260MiB / 7982MiB | 0%默认| + ------------------------------- + ----------------- ----- + ---------------------- +

+ ---------------------------------------------- ------------------------------- + |进程:GPU内存| | GPU PID类型进程名称用法| | 0 5282 C ... / anaconda3 / envs / deeplearning / bin / python 1249MiB | + ------------------------------------------------- ---------------------------- +

谢谢。

0 个答案:

没有答案