diff --git a/docker/Dockerfile.deepep b/docker/Dockerfile.deepep
index bd32e37d41b..e54e9df8d24 100644
--- a/docker/Dockerfile.deepep
+++ b/docker/Dockerfile.deepep
@@ -71,7 +71,7 @@ RUN NVSHMEM_SHMEM_SUPPORT=0 \
 
 WORKDIR /sgl-workspace/DeepEP
 ENV NVSHMEM_DIR=/sgl-workspace/nvshmem/install
-RUN NVSHMEM_DIR=/sgl-workspace/nvshmem/install python setup.py install
+RUN NVSHMEM_DIR=/sgl-workspace/nvshmem/install pip install .
 
 # Set workspace
 WORKDIR /sgl-workspace