Add cuda version to nccl

2022-04-28 15:10:26 +08:00 · 2022-04-28 15:10:26 +08:00 · fcabea5c07
parent aca75c0200
commit fcabea5c07
1 changed files with 2 additions and 1 deletions
--- a/cmake/external_libs/nccl.cmake
+++ b/cmake/external_libs/nccl.cmake
@ -6,9 +6,10 @@ else()
    set(MD5 "073b19899f374c5ba07d2db02dc38f9f")
 endif()

+find_package(CUDA REQUIRED)
 set(nccl_CFLAGS "-D_FORTIFY_SOURCE=2 -O2")
 mindspore_add_pkg(nccl
-        VER 2.7.6-1
+        VER 2.7.6-1-${CUDA_VERSION}
        LIBS nccl
        URL ${REQ_URL}
        MD5 ${MD5}