diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index e09429bc42957562423681fc4b0a1a3ea70e85cb..75f4f192444949381397fa4e78c705009e755692 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -11,7 +11,7 @@ elseif(NEW_RELEASE_ALL) add_definitions(-DNEW_RELEASE_ALL) set(paddle_known_gpu_archs "35 50 52 60 61 70 75 80 86") set(paddle_known_gpu_archs10 "35 50 52 60 61 70 75") - set(paddle_known_gpu_archs11 "35 50 60 61 70 75 80") + set(paddle_known_gpu_archs11 "50 60 61 70 75 80") elseif(NEW_RELEASE_PYPI) message("Using New Release Strategy - Cubin Packge") add_definitions(-DNEW_RELEASE_PYPI)