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'):