diff --git a/paddle/fluid/platform/cudnn_workspace_helper.h b/paddle/fluid/platform/cudnn_workspace_helper.h index 58f76e3128..29abdc72e2 100644 --- a/paddle/fluid/platform/cudnn_workspace_helper.h +++ b/paddle/fluid/platform/cudnn_workspace_helper.h @@ -17,7 +17,7 @@ namespace paddle { namespace platform { -static constexpr int kDefaultConvWorkspaceSizeLimitMB = 4096; +static constexpr int kDefaultConvWorkspaceSizeLimitMB = 512; } // namespace platform } // namespace paddle