set_config_gpu_low_memory {aifeducation} | R Documentation |
Setting gpus' memory usage
Description
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:
AifeducationConfiguration
,
aifeducation_config
,
check_aif_py_modules()
,
install_py_modules()
,
set_config_cpu_only()
,
set_config_os_environ_logger()
,
set_config_tf_logger()
,
set_transformers_logger()
[Package aifeducation version 0.3.3 Index]