diff --git a/configs/milvus.yaml b/configs/milvus.yaml index 21c20ec3eb..6d43cc0a24 100644 --- a/configs/milvus.yaml +++ b/configs/milvus.yaml @@ -1026,5 +1026,5 @@ trace: #milvus will automatically initialize half of the available GPU memory, #maxMemSize will the whole available GPU memory. gpu: - initMemSize: # Gpu Memory Pool init size - maxMemSize: # Gpu Memory Pool Max size + initMemSize: 2048 # Gpu Memory Pool init size + maxMemSize: 4096 # Gpu Memory Pool Max size diff --git a/pkg/util/paramtable/component_param.go b/pkg/util/paramtable/component_param.go index b68d2b7eaa..6c84430fa0 100644 --- a/pkg/util/paramtable/component_param.go +++ b/pkg/util/paramtable/component_param.go @@ -922,18 +922,20 @@ type gpuConfig struct { func (t *gpuConfig) init(base *BaseTable) { t.InitSize = ParamItem{ - Key: "gpu.initMemSize", - Version: "2.3.4", - Doc: `Gpu Memory Pool init size`, - Export: true, + Key: "gpu.initMemSize", + Version: "2.3.4", + Doc: `Gpu Memory Pool init size`, + Export: true, + DefaultValue: "2048", } t.InitSize.Init(base.mgr) t.MaxSize = ParamItem{ - Key: "gpu.maxMemSize", - Version: "2.3.4", - Doc: `Gpu Memory Pool Max size`, - Export: true, + Key: "gpu.maxMemSize", + Version: "2.3.4", + Doc: `Gpu Memory Pool Max size`, + Export: true, + DefaultValue: "4096", } t.MaxSize.Init(base.mgr) }