specify cuda arch when dectected fail (#26420)

specify cuda arch when dectected fail

specify cuda arch when dectected fail (#26420)
specify cuda arch when dectected fail
62bd7ba1 · Zhou Wei · GitHub · f00f982a · 62bd7ba1
隐藏空白更改
内联并排

Showing with 4 addition and 0 deletion

cmake/cuda.cmake cmake/cuda.cmake +4 -0

未找到文件。
--- 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()