diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index 87b943abd0106df3c659e8bf59a2be364ebe797a..63dbd0e9db4abfb2f783c1f1c1460a29e314828e 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -6,6 +6,7 @@ if(WITH_NV_JETSON) add_definitions(-DWITH_NV_JETSON) set(paddle_known_gpu_archs "53 62 72") set(paddle_known_gpu_archs10 "53 62 72") + set(paddle_known_gpu_archs11 "53 62 72") elseif(NEW_RELEASE_ALL) message("Using New Release Strategy - All Arches Packge") add_definitions(-DNEW_RELEASE_ALL)