Merge pull request #272 from sony/fix/20210120-fix-failed-to-compile-…

…nnabla-cuda-wheel Fix failed to compile nnabla cuda wheel.
sony · Jan 27, 2021 · 50d2fe5 · 50d2fe5
2 parents 6007ce1 + e5eebed
commit 50d2fe5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -210,6 +210,8 @@ def cuda_config(root_dir, cuda_lib, ext_opts, lib_dirs):
                 ls = l.strip().decode('ascii').split()
                 if len(ls) >= 3:
                     libname = ls[0]
+                    if libname == "libcuda.so.1":
+                        continue
                     libfile = ls[2]
 
                     # Copy libraries into WHL file.