Support cuda 11 with jetson (#47741)

14c95700 · chalsliu · GitHub · c3a69111 · 14c95700
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

cmake/cuda.cmake cmake/cuda.cmake +1 -0

未找到文件。
--- a/cmake/cuda.cmake
+++ b/cmake/cuda.cmake
@@ -6,6 +6,7 @@ if(WITH_NV_JETSON)
  add_definitions(-DWITH_NV_JETSON)
  set(paddle_known_gpu_archs "53 62 72")
  set(paddle_known_gpu_archs10 "53 62 72")
+  set(paddle_known_gpu_archs11 "53 62 72")
 elseif(NEW_RELEASE_ALL)
  message("Using New Release Strategy - All Arches Packge")
  add_definitions(-DNEW_RELEASE_ALL)