diff --git a/paddle/fluid/inference/api/demo_ci/CMakeLists.txt b/paddle/fluid/inference/api/demo_ci/CMakeLists.txt
index c02fcd078132126a8f2306083ff69e0d15170c76..547e265d2fdb51fbdc22c703206ba55d84e4316f 100644
--- a/paddle/fluid/inference/api/demo_ci/CMakeLists.txt
+++ b/paddle/fluid/inference/api/demo_ci/CMakeLists.txt
@@ -85,7 +85,7 @@ if(WITH_GPU)
     set(CUDA_LIB "/usr/local/cuda/lib64/" CACHE STRING "CUDA Library")
   else()
     set(CUDA_LIB "" CACHE STRING "CUDA_LIB")
-    if("${TENSORRT_ROOT}" STREQUAL "")
+    if("${CUDA_LIB}" STREQUAL "")
       if(DEFINED ENV{CUDA_PATH})
         set(CUDA_LIB "$ENV{CUDA_PATH}\\lib\\x64")
       else()
diff --git a/paddle/fluid/inference/tests/infer_ut/CMakeLists.txt b/paddle/fluid/inference/tests/infer_ut/CMakeLists.txt
index 5c17e2d62d37d141d3305a6f164daee34299077d..ad7ef0c04ce67d705e1b4639f4bd92f505caec26 100644
--- a/paddle/fluid/inference/tests/infer_ut/CMakeLists.txt
+++ b/paddle/fluid/inference/tests/infer_ut/CMakeLists.txt
@@ -84,7 +84,7 @@ if(WITH_GPU)
     set(CUDA_LIB "/usr/local/cuda/lib64/" CACHE STRING "CUDA Library")
   else()
     set(CUDA_LIB "" CACHE STRING "CUDA_LIB")
-    if("${TENSORRT_ROOT}" STREQUAL "")
+    if("${CUDA_LIB}" STREQUAL "")
       if(DEFINED ENV{CUDA_PATH})
         set(CUDA_LIB "$ENV{CUDA_PATH}\\lib\\x64")
       else()