检查失败：错误== cudaSuccess（2对0）内存不足

Question

我正在尝试在gpu上运行pycaffe的神经网络。

这是我第一次调用脚本时的工作原理。 当我第二次运行相同的脚本时，CUDA会在标题中抛出错误。

批量大小为1，此刻图像大小为243x322，gpu有8GB RAM。

我想我错过了一个重置​​内存的命令？

非常感谢！

编辑：

也许我应该澄清一些事情：我在Windows上运行caffe。

当我使用python script.py调用脚本时，进程终止并释放gpu内存，因此可以正常工作。

使用我用于调试的ipython，GPU内存确实没有被释放（一次通过后，8 bg中的6个正在使用中，感谢nvidia-smi建议！）

所以，我正在寻找的是一个我可以在pyhton中调用的命令，其中包括：

运行网络

处理图像输出

免费的gpu内存

Answer 1

你的GPU内存没有被释放。当前一个进程停止但未终止时会发生这种情况。请参阅我的回答here。

Answer 2

当GPU中的内存不足时会发生这种情况。你确定你正确地停止了第一个脚本吗？检查系统上的正在运行的进程（ubuntu中的ps -A），看看python脚本是否仍在运行。杀死它，如果是的话。您还应该检查GPU中使用的内存（nvidia-smi）。