我有一个脚本来训练CNN,在tensorflow 2.0.0上使用keras,当我使用pycharm时效果很好,但是当我从注释行(python train.py)运行它时,由于分配问题而崩溃:enter image description here我尝试了worker = 1,而use_multiprocessing = False是fit_generator,它没有帮助。知道为什么会这样吗?终端中缺少的pycharm会发生什么?
我正在NVIDIA tf 2.0.0 docker容器上运行。
[如果通过'命令行'是指在docker中运行,则您的tf容器无法像pycharm那样访问整个计算机RAM。 docker info
看一下内存,使用此处的https://dzone.com/articles/docker-container-resource-management-cpu-ram-and-i提示来调整需要的内存量