diff --git a/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h b/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h
index 193e08bdde558a616897cfef38fb446cd25ac6a2..555a83a0210f23824f77829410e0917a17455449 100644
--- a/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h
+++ b/paddle/fluid/platform/device/gpu/cuda/cuda_profiler.h
@@ -42,7 +42,7 @@ enum class NvtxRangeColor : uint32_t {
 };
 
 void CudaNvtxRangePush(const std::string& name,
-                       const NvtxRangeColor color = NvtxRangeColor::Blue);
+                       const NvtxRangeColor color = NvtxRangeColor::Green);
 
 void CudaNvtxRangePop();
 #endif
diff --git a/paddle/fluid/pybind/pybind.cc b/paddle/fluid/pybind/pybind.cc
index c37839a7c6509573fa2f03abb999ddf1b4147dd9..c5898b9762450f2a62176eb0540e79c5262a1cd2 100644
--- a/paddle/fluid/pybind/pybind.cc
+++ b/paddle/fluid/pybind/pybind.cc
@@ -2063,7 +2063,9 @@ All parameter, weight, gradient are variables in Paddle.
   m.def("nvprof_init", platform::CudaProfilerInit);
   m.def("nvprof_start", platform::CudaProfilerStart);
   m.def("nvprof_stop", platform::CudaProfilerStop);
-  m.def("nvprof_nvtx_push", platform::CudaNvtxRangePush);
+  m.def("nvprof_nvtx_push", [](const std::string &name) {
+    platform::CudaNvtxRangePush(name, platform::NvtxRangeColor::Green);
+  });
   m.def("nvprof_nvtx_pop", platform::CudaNvtxRangePop);
   m.def("nvprof_enable_record_event", platform::NvprofEnableRecordEvent);
   m.def("nvprof_disable_record_event", platform::NvprofDisableRecordEvent);