Versions Compared

Old Version 20

changes.mady.by.user Unknown User (davidsch)

Saved on

compared with

New Version 21

changes.mady.by.user Unknown User (davidsch)

Saved on

Key

  • This line was added.
  • This line was removed.
  • Formatting was changed.

...

will start an interactive ipython session where tensorflow will only see device 1. Tensorflow will call the one device it sees device 0.

 

Tensorflow: Limit GPU Memory

Memory

on a Card

With tensorflow, you can write your code to only grab the GPU memory that you need:

with tf.device('/gpu:0'): # this with statement may not be necessary
    config = tf.ConfigProto()
    config.gpu_options.allow_growth = True
    config.allow_soft_placement=True
    with tf.Session(config=config) as sess:
         # now your program, all variables will default to going on the GPU, and
         # any that shouldn't go on a GPU will be put on the CPU.

 

Configuration Subject to Change

...