diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 0c1650f9b..fbbf7d3a6 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -14,6 +14,8 @@ export USE_STATIC_NCCL=1 export ATEN_STATIC_CUDA=1 export USE_CUDA_STATIC_LINK=1 export INSTALL_TEST=0 # dont install test binaries into site-packages +# Set RPATH instead of RUNPATH when using patchelf to avoid LD_LIBRARY_PATH override +export FORCE_RPATH="--force-rpath" # Keep an array of cmake variables to add to if [[ -z "$CMAKE_ARGS" ]]; then