ci: fix nccl related build error (#497)

issue: 'nccl.__file__' returns None in new version.

Author: guocuimi

cmake/FindNCCL.cmake

@@ -28,19 +28,6 @@ find_path(NCCL_INCLUDE_DIR
 )
 mark_as_advanced(NCCL_INCLUDE_DIR)
 
-if (USE_STATIC_NCCL)
-  MESSAGE(STATUS "USE_STATIC_NCCL is set. Linking with static NCCL library.")
-  SET(NCCL_LIBNAME "nccl_static")
-  if (NCCL_VERSION)  # Prefer the versioned library if a specific NCCL version is specified
-    set(CMAKE_FIND_LIBRARY_SUFFIXES ".a.${NCCL_VERSION}" ${CMAKE_FIND_LIBRARY_SUFFIXES})
-  endif()
-else()
-  SET(NCCL_LIBNAME "nccl")
-  if (NCCL_VERSION)  # Prefer the versioned library if a specific NCCL version is specified
-    set(CMAKE_FIND_LIBRARY_SUFFIXES ".so.${NCCL_VERSION}" ${CMAKE_FIND_LIBRARY_SUFFIXES})
-  endif()
-endif()
-
 # Read version from header
 if(EXISTS "${NCCL_INCLUDE_DIR}/nccl.h")
   file(READ ${NCCL_INCLUDE_DIR}/nccl.h NCCL_HEADER_CONTENTS)
@@ -67,7 +54,7 @@ if(NCCL_HEADER_CONTENTS)
 endif()
 
 find_library(NCCL_LIBRARY
-  NAMES ${NCCL_LIBNAME}
+  NAMES nccl
   HINTS
     $ENV{NCCL_ROOT}
     $ENV{CUDA_HOME}

setup.py

@@ -25,15 +25,6 @@ def get_torch_root():
         return None
 
 
-def get_nccl_root():
-    try:
-        from nvidia import nccl
-
-        return str(Path(nccl.__file__).parent)
-    except ImportError:
-        return None
-
-
 def get_cuda_version():
     import torch.utils.cpp_extension as torch_cpp_ext
 
@@ -215,11 +206,6 @@ def build_extension(self, ext: CMakeExtension):
             )
         env["LIBTORCH_ROOT"] = LIBTORCH_ROOT
 
-        NCCL_ROOT = get_nccl_root()
-        if NCCL_ROOT is not None:
-            env["NCCL_ROOT"] = NCCL_ROOT
-            env["NCCL_VERSION"] = "2"
-
         # print cmake args
         print("CMake Args: ", cmake_args)
         print("Env: ", env)