From 212bcc9967a9bcfa18a103515aa9468020a5826e Mon Sep 17 00:00:00 2001 From: Tianlei Wu Date: Thu, 15 Aug 2024 00:03:10 -0700 Subject: [PATCH] Exclude cuDNN 9 and CUDA 12 DLLs from manylinux wheel (#21738) ### Description Exclude cuDNN 9 and CUDA 12 DLLs from manylinux wheel to reduce python package size. ### Motivation and Context The 1.20.0 ort-nightly-gpu python wheels on linux are suddenly > 800 MB in size. The wheels built on 1.19 release branch have a size of around 220 MB. The size change is caused by https://github.com/microsoft/onnxruntime/pull/19470. --- setup.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/setup.py b/setup.py index 1fa297e22acd9..96b9db5695377 100644 --- a/setup.py +++ b/setup.py @@ -208,6 +208,16 @@ def run(self): "libcufft.so.10", "libcufft.so.11", "libcurand.so.10", + "libcudnn_adv.so.9", + "libcudnn_cnn.so.9", + "libcudnn_engines_precompiled.so.9", + "libcudnn_engines_runtime_compiled.so.9", + "libcudnn_graph.so.9", + "libcudnn_heuristic.so.9", + "libcudnn_ops.so.9", + "libnvJitLink.so.12", + "libnvrtc.so.12", + "libnvrtc-builtins.so.12", ] rocm_dependencies = [ "libamd_comgr.so.2",