Add CUDA_ARCHITECTURES to CMakeLists.txt

CMakeLists.txt

@@ -36,3 +36,7 @@ set(SOURCES ${SOURCES_CU} ${SOURCES_CXX})
 set(HEADERS ${HEADERS_CU} ${HEADERS_CXX})
 
 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