Differences
This shows you the differences between two versions of the page.
Both sides previous revision Previous revision Next revision | Previous revision | ||
computers [2019/07/03 23:38] mai [Backup1] |
computers [2019/07/03 23:44] (current) mai |
||
---|---|---|---|
Line 43: | Line 43: | ||
| 1 93895 C python3 10863MiB | | | 1 93895 C python3 10863MiB | | ||
+-----------------------------------------------------------------------------+ | +-----------------------------------------------------------------------------+ | ||
+ | </code> | ||
+ | |||
+ | Par défaut, Tensorflow alloue l'ensemble de la mémoire GPU au programme, même s'il n'en nécessite que 10%. | ||
+ | Il existe une option permettant d'allouer seulement la mémoire GPU nécessaire. | ||
+ | Il faut ajouter les lignes suivantes: | ||
+ | <code> | ||
+ | config = tf.ConfigProto() | ||
+ | config.gpu_options.allow_growth = True | ||
+ | sess = tf.Session(config=config) | ||
+ | |||
</code> | </code> | ||