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
10
10
mkdir -p $WORKSPACE
11
11
fi
12
12
13
+ # configurable pip command (default: pip3)
14
+ PIP_CMD=${PIP_CMD:- pip3}
15
+ CUDA_HOME=${CUDA_HOME:-/ usr/ local/ cuda}
16
+
13
17
# install dependencies if not installed
14
- pip3 install cmake torch ninja
18
+ $PIP_CMD install cmake torch ninja
15
19
16
20
# build nvshmem
17
21
pushd $WORKSPACE
@@ -110,15 +114,13 @@ clone_repo() {
110
114
pushd $WORKSPACE
111
115
clone_repo " https://github.com/ppl-ai/pplx-kernels" " pplx-kernels" " setup.py" " c336faf"
112
116
cd 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 .
116
118
popd
117
119
118
120
# build and install deepep, require pytorch installed
119
121
pushd $WORKSPACE
120
122
clone_repo " https://github.com/deepseek-ai/DeepEP" " DeepEP" " setup.py" " e3908bf"
121
123
cd DeepEP
122
124
export NVSHMEM_DIR=$WORKSPACE /nvshmem_install
123
- PIP_NO_BUILD_ISOLATION=0 pip install -vvv -e .
125
+ $PIP_CMD install --no-build-isolation - vvv -e .
124
126
popd
You can’t perform that action at this time.
0 commit comments