产生CUDA Memory的原因很多,在这里我只说一下在我PC上训练模型时产生该问题的解决方式。
我在训练模型的时候,观察到GPU的利用率为0,却仍报这种错误。我尝试了以下操作,发现不再报错。
在控制面板的系统中找到编辑系统环境变量(或者直接搜索这个编辑系统环境变量)。打开以后,点击高级中的设置:
点击高级里面的更改:
把C盘和装有python环境的盘(我的是F盘)(最好这两个盘都设置一下大小),点击自定义大小,设置成一个较大的值:
点击确定,在重新运行程序,就不再报错了。
由于GPU的使用按理说应该不会用到虚拟内存,但是不知道为啥调整这个虚拟内存的分配大小就能够解决问题。