我正在使用GPU和Nvidia Quadro M4000 8GB显卡进行张量调整。 我遇到了内存错误问题。我尝试过使用内存增长方法,但仍然面临同样的问题。
ResourceExhaustedError: OOM when allocating tensor with shape[1,64,1228,1029]
这里形状[1,64,1228,1029]位于第一个汇集层之后
Windows 7上已经安装了所有内容:
答案 0 :(得分:0)
好吧,因为这只是323兆字节,所以只分配这个张量不应该是唯一的问题。
尝试查看您在程序中分配的其他张量。很可能你正在分配许多这样的大小,或者你的程序中的许多其他地方。
您可能遇到的另一个问题是您正在同时运行多个程序。尝试在Windows中运行nvidia-smi
等效项,以查看您的卡目前使用了多少内存。