在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

修改以后我的问题解决了

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐