Support cuda 11 on jetson (#47742)

b3e45714 · chalsliu · GitHub · d0e19af3 · b3e45714
隐藏空白更改
内联并排

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)