follow comments

8 years ago · b3ab15a7ab
parent 83d0016f54
commit b3ab15a7ab
1 changed files with 2 additions and 1 deletions
--- 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();