File tree Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Expand file tree Collapse file tree 1 file changed +7
-5
lines changed Original file line number Diff line number Diff line change @@ -10,8 +10,12 @@ if [ ! -d "$WORKSPACE" ]; then
1010 mkdir -p $WORKSPACE
1111fi
1212
13+ # configurable pip command (default: pip3)
14+ PIP_CMD=${PIP_CMD:- pip3}
15+ CUDA_HOME=${CUDA_HOME:-/ usr/ local/ cuda}
16+
1317# install dependencies if not installed
14- pip3 install cmake torch ninja
18+ $PIP_CMD install cmake torch ninja
1519
1620# build nvshmem
1721pushd $WORKSPACE
@@ -110,15 +114,13 @@ clone_repo() {
110114pushd $WORKSPACE
111115clone_repo " https://github.com/ppl-ai/pplx-kernels" " pplx-kernels" " setup.py" " c336faf"
112116cd pplx-kernels
113- # see https://github.com/pypa/pip/issues/9955#issuecomment-838065925
114- # PIP_NO_BUILD_ISOLATION=0 disables build isolation
115- PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
117+ $PIP_CMD install --no-build-isolation -vvv -e .
116118popd
117119
118120# build and install deepep, require pytorch installed
119121pushd $WORKSPACE
120122clone_repo " https://github.com/deepseek-ai/DeepEP" " DeepEP" " setup.py" " e3908bf"
121123cd DeepEP
122124export NVSHMEM_DIR=$WORKSPACE /nvshmem_install
123- PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
125+ $PIP_CMD install --no-build-isolation - vvv -e .
124126popd
You can’t perform that action at this time.
0 commit comments