From 59f6271da100c320d004032b7415c12fc50a1269 Mon Sep 17 00:00:00 2001 From: Matthew Nelson - Silverark Date: Thu, 5 Aug 2021 22:15:03 +0100 Subject: [PATCH] cuda_mem_limit has been renamed to gpu_mem_limit (#8284) `cuda_mem_limit` was renamed to `gpu_mem_limit` in nighlty build `nightly-1.7.0.dev202104231`. https://github.com/microsoft/onnxruntime/issues/7457 --- .../reference/execution-providers/CUDA-ExecutionProvider.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/docs/reference/execution-providers/CUDA-ExecutionProvider.md b/docs/reference/execution-providers/CUDA-ExecutionProvider.md index 23f92b3195..534dc235ef 100644 --- a/docs/reference/execution-providers/CUDA-ExecutionProvider.md +++ b/docs/reference/execution-providers/CUDA-ExecutionProvider.md @@ -45,7 +45,7 @@ The device ID. Default value: 0 -### cuda_mem_limit +### gpu_mem_limit The size limit of the device memory arena in bytes. This size limit is only for the execution provider's arena. The total device memory usage may be higher. s: max value of C++ size_t type (effectively unlimited) @@ -88,7 +88,7 @@ providers = [ ('CUDAExecutionProvider', { 'device_id': 0, 'arena_extend_strategy': 'kNextPowerOfTwo', - 'cuda_mem_limit': 2 * 1024 * 1024 * 1024, + 'gpu_mem_limit': 2 * 1024 * 1024 * 1024, 'cudnn_conv_algo_search': 'EXHAUSTIVE', 'do_copy_in_default_stream': True, }), @@ -106,7 +106,7 @@ OrtSessionOptions* session_options = /* ... */; OrtCUDAProviderOptions options; options.device_id = 0; options.arena_extend_strategy = 0; -options.cuda_mem_limit = 2 * 1024 * 1024 * 1024; +options.gpu_mem_limit = 2 * 1024 * 1024 * 1024; options.cudnn_conv_algo_search = OrtCudnnConvAlgoSearch::EXHAUSTIVE; options.do_copy_in_default_stream = 1;