physical_devices = tf.config.experimental.list_physical_devices('GPU')

assert len(physical_devices) > 0, "Not enough GPU hardware devices available"

config = tf.config.experimental.set_memory_growth(physical_devices[0], True)

$ export TF_FORCE_GPU_ALLOW_GROWTH=true

Run the following

rm -rf ~/.nv/ 

​

and set 

config.gpu_options.allow_growth = True

physical_devices = tf.config.experimental.list_physical_devices('GPU')

assert len(physical_devices) > 0, "Not enough GPU hardware devices available"

config = tf.config.experimental.set_memory_growth(physical_devices[0], True)

​