diff --git a/CMakeLists.txt b/CMakeLists.txt
index 2437aed419b865288e218e864b2bad3a49000d4c..b475760cb44e244d279c4f6e3119de99b87dfe86 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -35,4 +35,8 @@ endif(INCLUDE_CUDA)
 set(SOURCES ${SOURCES_CU} ${SOURCES_CXX})
 set(HEADERS ${HEADERS_CU} ${HEADERS_CXX})
 
-add_library(memproc STATIC ${HEADERS} ${SOURCES})
\ No newline at end of file
+add_library(memproc STATIC ${HEADERS} ${SOURCES})
+
+if(INCLUDE_CUDA)
+    set_target_properties(memproc PROPERTIES CUDA_ARCHITECTURES "60;80")
+endif(INCLUDE_CUDA)
\ No newline at end of file