2016-07-27 22 views
6

Ich habe einige Fragen über GPU-Speicher mit Tensorflow gesehen, aber ich habe es auf einem Pine64 ohne GPU-Unterstützung installiert.Limit Tensorflow CPU und Speicherverbrauch

Das bedeutet, dass ich es mit sehr begrenzten Ressourcen (nur CPU und RAM) laufen und Tensorflow scheint alles zu wollen, meine Maschine vollständig einzufrieren.


Gibt es einen Weg, um die Menge an Rechenleistung und Speicher zu begrenzen zugeordnet Tensorflow? Ähnlich wie Bazels eigene --local_resources Flagge?

Antwort

9

Dies wird eine Sitzung erstellen, die eine op zu einem Zeitpunkt ausgeführt wird, und nur ein Thread pro op

sess = tf.Session(
    tf.ConfigProto(inter_op_parallelism_threads=1, 
        intra_op_parallelism_threads=1)) 

Nicht sicher Speicher zu begrenzen, so scheint es auf Nachfrage zugeteilt werden, ich habe TensorFlow freeze hatte mein Computer, wenn mein Netzwerk 100 GB RAM wollte, so war meine Lösung, Netzwerke, die weniger RAM benötigen

+0

Das warf mir die Ausnahme 'TypeError: Ziel muss eine Zeichenfolge sein, aber Exception AttributeError: Das' Session '-Objekt hat kein Attribut' _session '"in > ignored', aber Hinzufügen 'Config' Schlüsselwort (d. h. 'sess = tf.Session (config = tf.ConfigProto (inter_op_parallelism_threads = 1, intra_op_parallelism_threads = 1))') hat das Problem behoben. –