diff --git a/paddle/platform/gpu_info.h b/paddle/platform/gpu_info.h index 6a998382419376384a5d592efcb2bccbc590ce61..ed2420b8740e583d307f6836a70fe7e1c780e28b 100644 --- a/paddle/platform/gpu_info.h +++ b/paddle/platform/gpu_info.h @@ -24,7 +24,8 @@ namespace paddle { namespace platform { //! Environment variable: fraction of GPU memory to use on each device. -const std::string kEnvFractionGpuMemoryToUse = "FRACTION_GPU_MEMORY_TO_USE"; +const std::string kEnvFractionGpuMemoryToUse = + "PADDLE_FRACTION_GPU_MEMORY_TO_USE"; //! Get the total number of GPU devices in system. int GetDeviceCount();