specify cuda arch when dectected fail (#26420)

specify cuda arch when dectected fail
5 years ago · 62bd7ba13c
parent f00f982a02
commit 62bd7ba13c
1 changed files with 4 additions and 0 deletions
--- a/cmake/cuda.cmake
+++ b/cmake/cuda.cmake
@ -61,6 +61,10 @@ function(detect_installed_gpus out_variable)
  if(NOT CUDA_gpu_detect_output)
    message(STATUS "Automatic GPU detection failed. Building for all known architectures.")
    set(${out_variable} ${paddle_known_gpu_archs} PARENT_SCOPE)
+    #Todo: fix Automatic GPU detection failed on windows
+    if(WIN32)
+      set(${out_variable} "61 75" PARENT_SCOPE)
+    endif()
  else()
    set(${out_variable} ${CUDA_gpu_detect_output} PARENT_SCOPE)
  endif()