diff --git a/CMakeLists.txt b/CMakeLists.txt
index b475760cb44e244d279c4f6e3119de99b87dfe86..ad2872e3bb5ab64e70a171699acecffaecb7bf3d 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -10,8 +10,7 @@ message(STATUS "memory processing: GPU build in mode is ${INCLUDE_CUDA}")
 
 if(INCLUDE_CUDA)
     enable_language(CUDA)
-    # find_package(CUDA REQUIRED)
-    # include_directories(${CUDA_INCLUDE_DIRS})
+    include_directories(${CMAKE_CUDA_TOOLKIT_INCLUDE_DIRECTORIES})
 
     # -DCMAKE_CUDA_ARCHITECTURES=native -- Compile for the architecture(s) of the host's GPU(s)
 endif(INCLUDE_CUDA)