|
|
@ -20,8 +20,11 @@ limitations under the License. */
|
|
|
|
#include "paddle/fluid/platform/enforce.h"
|
|
|
|
#include "paddle/fluid/platform/enforce.h"
|
|
|
|
|
|
|
|
|
|
|
|
DEFINE_double(fraction_of_gpu_memory_to_use, 0.92,
|
|
|
|
DEFINE_double(fraction_of_gpu_memory_to_use, 0.92,
|
|
|
|
"Default use 92% of GPU memory for PaddlePaddle,"
|
|
|
|
"Allocate a trunk of gpu memory that is this fraction of the "
|
|
|
|
"reserve the rest for page tables, etc");
|
|
|
|
"total gpu memory size. Future memory usage will be allocated "
|
|
|
|
|
|
|
|
"from the trunk. If the trunk doesn't have enough gpu memory, "
|
|
|
|
|
|
|
|
"additional trunks of the same size will be requested from gpu "
|
|
|
|
|
|
|
|
"until the gpu has no memory left for another trunk.");
|
|
|
|
|
|
|
|
|
|
|
|
namespace paddle {
|
|
|
|
namespace paddle {
|
|
|
|
namespace platform {
|
|
|
|
namespace platform {
|
|
|
|