fix bug in gpu default memory allocating policy (#6268)

7 years ago · 96a5f96cc1
parent 70ee79c256
commit 96a5f96cc1
1 changed files with 2 additions and 2 deletions
--- a/paddle/platform/gpu_info.cc
+++ b/paddle/platform/gpu_info.cc
@ -18,8 +18,8 @@ limitations under the License. */

 #include "paddle/platform/enforce.h"

-DEFINE_double(fraction_of_gpu_memory_to_use, 0.95,
-              "Default use 95% of GPU memory for PaddlePaddle,"
+DEFINE_double(fraction_of_gpu_memory_to_use, 0.92,
+              "Default use 92% of GPU memory for PaddlePaddle,"
              "reserve the rest for page tables, etc");

 namespace paddle {