Add CUDA_ARCHITECTURES to CMakeLists.txt

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