From 0d30b42c7bc89ec07611cd763e170e68d218cb35 Mon Sep 17 00:00:00 2001 From: Changming Sun Date: Mon, 24 Feb 2020 09:51:40 -0800 Subject: [PATCH] Fix python gpu build (#3081) The problem was introduced in #2999. After that, we can't publish our python wheels to pypi because the package size is too big. --- setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/setup.py b/setup.py index 4af75ccc87..34d77b699f 100644 --- a/setup.py +++ b/setup.py @@ -92,7 +92,7 @@ try: logger.info('copying %s -> %s', source, dest) copyfile(source, dest) result = subprocess.run(['patchelf', '--print-needed', dest], check=True, stdout=subprocess.PIPE, universal_newlines=True) - cuda_dependencies = ['libcublas.so', 'libcudnn.so', 'libcudart.so'] + cuda_dependencies = ['libcublas.so', 'libcudnn.so', 'libcudart.so', 'libcurand.so'] to_preload = [] args = ['patchelf', '--debug'] for line in result.stdout.split('\n'):