在pycharm中运行的python脚本从命令行运行时出现内存分配错误

问题描述 投票:1回答:1

我有一个脚本来训练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容器上运行。

python tensorflow terminal pycharm multiprocessing
1个回答
0
投票

[如果通过'命令行'是指在docker中运行,则您的tf容器无法像pycharm那样访问整个计算机RAM。 docker info看一下内存,使用此处的https://dzone.com/articles/docker-container-resource-management-cpu-ram-and-i提示来调整需要的内存量

© www.soinside.com 2019 - 2024. All rights reserved.