chore: add dev-containers

huggingface · Dec 13, 2024 · 89e77c2 · 89e77c2
1 parent 51d064d
commit 89e77c2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/.devcontainer/Dockerfile_trtllm b/.devcontainer/Dockerfile_trtllm
@@ -53,7 +53,7 @@ RUN wget "https://download.open-mpi.org/release/open-mpi/v4.1/$OMPI_TARBALL_FILE
 
 # Install TensorRT
 FROM cuda-builder AS trt-builder
-COPY install_tensorrt.sh /opt/install_tensorrt.sh
+COPY backends/trtllm/scripts/install_tensorrt.sh /opt/install_tensorrt.sh
 RUN chmod +x /opt/install_tensorrt.sh && \
     /opt/install_tensorrt.sh
 

diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
@@ -2,9 +2,9 @@
 // README at: https://github.com/devcontainers/templates/tree/main/src/cpp
 {
   "name": "CUDA",
-//  "image": "nvidia/cuda:12.6.0-devel-ubuntu24.04",
   "build": {
-    "dockerfile": "Dockerfile_trtllm"
+    "dockerfile": "Dockerfile_trtllm",
+    "context": ".."
   },
   "remoteEnv": {
     "PATH": "${containerEnv:PATH}:/usr/local/cuda/bin",
@@ -15,8 +15,5 @@
     "jetbrains" : {
       "backend" : "CLion"
     }
-  },
-
-	// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
-	// "remoteUser": "root"
+  }
 }