MemoryError: Unable to allocate 6.73 GiB for an array with shape (3000, 3, 448, 448) and data type f

在ubuntu系统上得docker中运行pytorch代码，报出如下错误

MemoryError: Unable to allocate 6.73 GiB for an array with shape (3000, 3, 448, 448) and data type float32

主要是因为一次开辟得内存过大，此时
1、内存过载处理模式

cat /proc/sys/vm/overcommit_memory 

应该显示是0
2、切换到root身份并更改处理模式

echo 1 > /proc/sys/vm/overcommit_memory

修改以后我的问题解决了