Как ограничить использование памяти тензорным потоком? - PullRequest
0 голосов
/ 24 апреля 2018

Я новичок в tenorflow, я обнаружил, что он занимает слишком много памяти, когда я запускаю простой скрипт. Я не имею в виду память GPU, я просто обнаружил, что она стоит слишком дорого память .

Вот мой сценарий:

# -*- 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) 

Использование памяти вышеупомянутой программой python

Согласно моим наблюдениям, «import tenorflow as tf» занимает около 100 МБ, а tf.Session - другие.

Ну, интересно, есть ли способ его оптимизировать?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...