diff --git a/python/setup.py.in b/python/setup.py.in index c41025556377362c31bfa63a457500a033241d52..7ee37a7a21ff5eb2a123f8a2debad0ca5a963624 100644 --- a/python/setup.py.in +++ b/python/setup.py.in @@ -278,11 +278,12 @@ else: # copy the openblas.dll shutil.copy('${OPENBLAS_SHARED_LIB}', libs_path) package_data['paddle.libs'] += ['openblas' + ext_name] - elif os.name == 'posix' and platform.machine() == 'aarch64': + elif os.name == 'posix' and platform.machine() == 'aarch64' and '${OPENBLAS_LIB}'.endswith('so'): # copy the libopenblas.so on linux+aarch64 # special: core_noavx.so depends on 'libopenblas.so.0', not 'libopenblas.so' - shutil.copy('${OPENBLAS_LIB}' + '.0', libs_path) - package_data['paddle.libs'] += ['libopenblas.so.0'] + if os.path.exists('${OPENBLAS_LIB}' + '.0'): + shutil.copy('${OPENBLAS_LIB}' + '.0', libs_path) + package_data['paddle.libs'] += ['libopenblas.so.0'] if '${WITH_LITE}' == 'ON': shutil.copy('${LITE_SHARED_LIB}', libs_path)