diff --git a/iso/builder/build-nccl-tests.sh b/iso/builder/build-nccl-tests.sh index 5abc6f6..7014a02 100755 --- a/iso/builder/build-nccl-tests.sh +++ b/iso/builder/build-nccl-tests.sh @@ -118,6 +118,7 @@ else -gencode=arch=compute_100,code=sm_100" echo "gencode: sm_70..sm_100 (CUDA 12)" fi +LIBRARY_PATH="$NCCL_LIB_DIR${LIBRARY_PATH:+:$LIBRARY_PATH}" \ make MPI=0 \ NVCC="$NVCC" \ CUDA_HOME="$CUDA_HOME" \