本文共 602 字，大约阅读时间需要 2 分钟。

出现RuntimeError: CUDA error: out of memory解决办法记录

在训练或测试过程中出现错误如下：或者出现这样类似的错误：

RuntimeError: Error while calling cudaMalloc(&data, n) in file C:\Users\43196\Desktop\dlib-19.19\dlib\cuda\cuda_data_ptr.cpp:58. code: 2, reason: out of memory

原因： 由于GPU显存资源有限，训练输入的batchsize或者图片大小imagesize过大，会导致GPU爆满引发中断，会出现out of memory错误

解决方案：

测试时出现此问题解决方案： 在测试代码之前使用命令行： with torch.no_grad():

此外，可以使用终止进程释放内存。

watch -n 0.1 nvidia-smi

然后，杀死程序释放内存。

ps laf//看哪个pytorch 的 pid kill -9 pid //终止进程nvidia-smi//查看GPU使用情况

转载地址：http://uhtki.baihongyu.com/