Fix post-build-script (pytorch#8982)

(cherry picked from commit 29066f5)

Author: NicolasHug

packaging/post_build_script.sh

@@ -1,4 +1,10 @@
 #!/bin/bash
-LD_LIBRARY_PATH="/usr/local/lib:$CUDA_HOME/lib64:$LD_LIBRARY_PATH" python packaging/wheel/relocate.py
+set -euxo pipefail
+
+if [ -n "${CUDA_HOME:-}" ]; then
+    LD_LIBRARY_PATH="/usr/local/lib:${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}"
+fi
+
+python packaging/wheel/relocate.py
 
 pip install torchvision-extra-decoders

packaging/pre_build_script.sh

@@ -35,7 +35,7 @@ else
   conda install libwebp -yq
   conda install libjpeg-turbo -c pytorch
   yum install -y freetype gnutls
-  pip install auditwheel
+  pip install "auditwheel<6.3.0"
 fi
 
 pip install numpy pyyaml future ninja

packaging/wheel/relocate.py

@@ -15,7 +15,10 @@
 
 # Third party imports
 if sys.platform == "linux":
-    from auditwheel.lddtree import lddtree
+    try:
+        from auditwheel.lddtree import lddtree
+    except ImportError:
+        from auditwheel import lddtree
 
 
 ALLOWLIST = {