小编pea*_*eak的帖子

如何限制 tensorflow 内存使用?

我是 TensorFlow 的新手。当我运行一个简单的脚本时,我发现它占用了太多内存。我不是说 GPU 内存,我是说CPU内存。

这是我的脚本:

# -*- coding: utf-8 -*-

import time
import tensorflow as tf
tf_config = tf.ConfigProto()
tf_config.gpu_options.allow_growth = False
with tf.Session(config=tf_config) as sess:
    print('Listening.....')
    time.sleep(100) 
Run Code Online (Sandbox Code Playgroud)

上面python程序的内存使用情况

根据我的观察,'import tensorflow as tf' 需要大约 100MB,而 tf.Session 需要其他人。

嗯,我想知道有没有什么方法可以优化它?

python tensorflow

6
推荐指数
1
解决办法
2598
查看次数

标签 统计

python ×1

tensorflow ×1