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> | ||