kanshan/TensorRT-LLMs
1
0
Fork 0
mirror of https://github.com/NVIDIA/TensorRT-LLM.git synced 2026-01-13 22:18:36 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

infra: Devcontainer productivity improvements (#3075)

Browse Source 
* devcontainer improvements

Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>

* docker compose path

Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>

* updated some more devcontainer settings

Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>

* clean devcontainer name

Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>

---------

Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>
This commit is contained in:
Lucas Liebenwein 2025-04-02 14:23:38 -04:00 committed by GitHub
parent abcb0486dc
commit 5fc2f63fec
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 15 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
.devcontainer/clangd/devcontainer.json → .devcontainer/devcontainer.json
View File

@ -1,23 +1,30 @@
// For format details, see https://aka.ms/devcontainer.json. For config options, see the README at:
// https://github.com/microsoft/vscode-dev-containers/tree/v0.245.2/containers/docker-existing-dockerfile
{
"name": "TRT-LLM clangd Devcontainer",
"name": "TRT-LLM Devcontainer",
"dockerComposeFile": [
"../docker-compose.yml"
"docker-compose.yml"
],
"service": "tensorrt_llm-dev",
"remoteUser": "ubuntu",
"containerEnv": {
// "CCACHE_DIR" : "/home/coder/${localWorkspaceFolderBasename}/cpp/.ccache",
// "CCACHE_BASEDIR" : "/home/coder/${localWorkspaceFolderBasename}",
"HF_TOKEN": "${localEnv:HF_TOKEN}",
"HF_HOME": "/huggingface",
"HISTFILE": "${containerWorkspaceFolder}/.cache/._bash_history"
},
"workspaceFolder": "/workspaces/tensorrt_llm",
// "workspaceFolder": "/home/coder/${localWorkspaceFolderBasename}",
// "workspaceMount": "source=${localWorkspaceFolder},target=/home/coder/${localWorkspaceFolderBasename},type=bind,consistency=consistent",
"mounts": [
"source=${localEnv:HOME}/.cache/huggingface,target=/huggingface,type=bind", // HF cache
"source=/home/scratch.trt_llm_data/,target=/home/scratch.trt_llm_data/,type=bind,consistency=consistent"
],
// Note: sourcing .profile is required since we use a local user and the python interpreter is
// global (/usr/bin/python). In this case, pip will default to a local user path which is not
// by default in the PATH. In interactive devcontainer shells, .profile is sourced by default.
"postCreateCommand": "mkdir -p $HOME/.local/bin && source $HOME/.profile && pip install -r requirements-dev.txt && pre-commit install --install-hooks",
"customizations": {
"vscode": {
"extensions": [
@ -28,6 +35,8 @@
// Python
"ms-python.python",
"eeyore.yapf",
// AutoDeploy Linting and Formatting
"charliermarsh.ruff",
// Build Tools
// "ms-azuretools.vscode-docker",
// "ms-vscode.makefile-tools",
@ -48,8 +57,9 @@
// "clang-format.executable": "/usr/local/bin/clang-format",
"clangd.arguments": [
// "--compile-commands-dir=${workspaceFolder}/cpp/build_RelWithDebInfo"
"--compile-commands-dir=${workspaceFolder}"
]
"--compile-commands-dir=${workspaceFolder}/cpp/build"
],
"python.defaultInterpreterPath": "/usr/bin/python"
}
}
},

2
.devcontainer/docker-compose.yml
View File

@ -19,7 +19,7 @@ services:
reservations:
devices:
- driver: nvidia
count: 1
count: "all"
capabilities: [gpu]
volumes: