diff --git a/.ci/aarch64_linux/aarch64_wheel_ci_build.py b/.ci/aarch64_linux/aarch64_wheel_ci_build.py index d463f8474868..92dabf0fee48 100755 --- a/.ci/aarch64_linux/aarch64_wheel_ci_build.py +++ b/.ci/aarch64_linux/aarch64_wheel_ci_build.py @@ -99,10 +99,14 @@ def update_wheel(wheel_path, desired_cuda) -> None: if "126" in desired_cuda: libs_to_copy += [ "/usr/local/cuda/lib64/libnvrtc-builtins.so.12.6", + "/usr/local/cuda/lib64/libcufile.so.0", + "/usr/local/cuda/lib64/libcufile_rdma.so.1", ] elif "128" in desired_cuda: libs_to_copy += [ "/usr/local/cuda/lib64/libnvrtc-builtins.so.12.8", + "/usr/local/cuda/lib64/libcufile.so.0", + "/usr/local/cuda/lib64/libcufile_rdma.so.1", ] else: libs_to_copy += [