训练时出现的显存错误CUDA error: out of memory_cuda error out of memory

作者：盐析白兔 | 2024-03-11 21:36:38

踩

cuda error out of memory

RuntimeError: CUDA error: out of memory

训练时报错：
RuntimeError: CUDA error: out of memory

出现这种错误一般分两种情况：
第一种情况是你的显卡的显存确实不够，解决方法就是改动 bathsize 这些超参数试试，或者氪金买卡；

第二种情况是显卡的性能能满足，但是被其他的进程占用了，所以可分配的显存不足，解决方法就是结束这些占用显卡的进程从而使现存得到释放。具体地：
在Linux中，在终端输入

nvidia-smi
1

查看占用显存的PID进程号，然后输入

kill -9 -PID
1

即可释放现存继续训练。

在Windows中，直接打开任务管理器进行查看和结束相应的进程。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/盐析白兔/article/detail/220654