saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-01 03:45:06 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix python gpu build (#3081)

Browse source 
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:
Changming Sun 2020-02-24 09:51:40 -08:00 committed by GitHub
parent f367fd921c
commit 0d30b42c7b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
setup.py
View file

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