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.
This commit is contained in:
Родитель
f367fd921c
Коммит
0d30b42c7b
2
setup.py
2
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'):
|
||||
|
|
Загрузка…
Ссылка в новой задаче