Setting gpus' memory usage

This function changes the memory usage of the gpus to allow computations on machines with small memory. With this function, some computations of large models may be possible but the speed of computation decreases.

Usage

set_config_gpu_low_memory()

Value

This function does not return anything. It is used for its side effects.

Note

This function sets TF_GPU_ALLOCATOR to "cuda_malloc_async" and sets memory growth to TRUE.

See also

Other Installation and Configuration Tensorflow: set_config_cpu_only(), set_config_os_environ_logger(), set_config_tf_logger()