diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake index d571925b20616fe724583ec06343a4084269c858..9ff908a4c87d55e87468a06ae0e6085ac165a1b1 100644 --- a/cmake/cuda.cmake +++ b/cmake/cuda.cmake @@ -4,9 +4,9 @@ endif() set(paddle_known_gpu_archs "30 35 50 52 60 61 70") set(paddle_known_gpu_archs7 "30 35 50 52") -set(paddle_known_gpu_archs8 "30 35 50 52 60 61") -set(paddle_known_gpu_archs9 "30 35 50 52 60 61 62 70") -set(paddle_known_gpu_archs10 "30 35 50 52 60 61 62 70 75") +set(paddle_known_gpu_archs8 "30 35 50 52 53 60 61 62") +set(paddle_known_gpu_archs9 "30 35 50 52 53 60 61 62 70") +set(paddle_known_gpu_archs10 "30 35 50 52 53 60 61 62 70 72 75") ###################################################################################### # A function for automatic detection of GPUs installed (if autodetection is enabled)