diff --git a/.devcontainer/Dockerfile.trtllm b/.devcontainer/Dockerfile_trtllm
similarity index 96%
rename from .devcontainer/Dockerfile.trtllm
rename to .devcontainer/Dockerfile_trtllm
index 4cf85388..e7603e09 100644
--- a/.devcontainer/Dockerfile.trtllm
+++ b/.devcontainer/Dockerfile_trtllm
@@ -70,4 +70,6 @@ ENV PATH="/root/.cargo/bin:$PATH"
 RUN cargo install cargo-chef
 
 COPY --from=trt-builder /usr/local/tensorrt /usr/local/tensorrt
-COPY --from=mpi-builder /usr/local/mpi /usr/local/mpi
\ No newline at end of file
+COPY --from=mpi-builder /usr/local/mpi /usr/local/mpi
+
+ENV MPI_HOME=/usr/local/mpi
\ No newline at end of file
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
index 5a2f347d..08acb6b4 100644
--- a/.devcontainer/devcontainer.json
+++ b/.devcontainer/devcontainer.json
@@ -4,7 +4,7 @@
   "name": "CUDA",
 //  "image": "nvidia/cuda:12.6.0-devel-ubuntu24.04",
   "build": {
-    "dockerfile": "Dockerfile.trtllm"
+    "dockerfile": "Dockerfile_trtllm"
   },
   "remoteEnv": {
     "PATH": "${containerEnv:PATH}:/usr/local/cuda/bin",