From e734a2a085bfed665ac5d8223d991d4ba43b58e0 Mon Sep 17 00:00:00 2001 From: Lucas Wilkinson Date: Tue, 30 Sep 2025 19:38:34 -0400 Subject: [PATCH] [Misc] Make EP kernels install script support uv (#25785) Signed-off-by: Lucas Wilkinson Signed-off-by: yewentao256 --- tools/ep_kernels/install_python_libraries.sh | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/tools/ep_kernels/install_python_libraries.sh b/tools/ep_kernels/install_python_libraries.sh index 59bfe69dc0dd6..5a3d734190c1a 100644 --- a/tools/ep_kernels/install_python_libraries.sh +++ b/tools/ep_kernels/install_python_libraries.sh @@ -10,8 +10,12 @@ if [ ! -d "$WORKSPACE" ]; then mkdir -p $WORKSPACE fi +# configurable pip command (default: pip3) +PIP_CMD=${PIP_CMD:-pip3} +CUDA_HOME=${CUDA_HOME:-/usr/local/cuda} + # install dependencies if not installed -pip3 install cmake torch ninja +$PIP_CMD install cmake torch ninja # build nvshmem pushd $WORKSPACE @@ -110,9 +114,7 @@ clone_repo() { pushd $WORKSPACE clone_repo "https://github.com/ppl-ai/pplx-kernels" "pplx-kernels" "setup.py" "c336faf" cd pplx-kernels -# see https://github.com/pypa/pip/issues/9955#issuecomment-838065925 -# PIP_NO_BUILD_ISOLATION=0 disables build isolation -PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e . +$PIP_CMD install --no-build-isolation -vvv -e . popd # build and install deepep, require pytorch installed @@ -120,5 +122,5 @@ pushd $WORKSPACE clone_repo "https://github.com/deepseek-ai/DeepEP" "DeepEP" "setup.py" "e3908bf" cd DeepEP export NVSHMEM_DIR=$WORKSPACE/nvshmem_install -PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e . +$PIP_CMD install --no-build-isolation -vvv -e . popd