From 62bd7ba13c6c9f23dfeb55680999c30f4c8ab0b1 Mon Sep 17 00:00:00 2001
From: Zhou Wei <52485244+zhouwei25@users.noreply.github.com>
Date: Thu, 20 Aug 2020 15:46:33 +0800
Subject: [PATCH] specify cuda arch when dectected fail (#26420)

specify cuda arch when dectected fail
---
 cmake/cuda.cmake | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/cmake/cuda.cmake b/cmake/cuda.cmake
index bb92eae732e..b7a93cd9ee2 100644
--- 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()
-- 
GitLab