huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: fix dev containers

Browse Source
This commit is contained in:
Hugo Larcher 2024-12-02 15:30:59 +01:00
parent bad878bdd5
commit 668476a241
No known key found for this signature in database
GPG Key ID: 3DAF63124699CA2B
5 changed files with 26 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
backends/trtllm/.devcontainer/Dockerfile → .devcontainer/Dockerfile
View File

26
.devcontainer/devcontainer.json
View File

@ -0,0 +1,26 @@
// For format details, see https://aka.ms/devcontainer.json. For config options, see the
// README at: https://github.com/devcontainers/templates/tree/main/src/cpp
{
"name": "CUDA",
"image": "nvidia/cuda:12.6.0-devel-ubuntu24.04",
// "runArgs": [
// "--gpus=all"
// ],
"remoteEnv": {
"PATH": "${containerEnv:PATH}:/usr/local/cuda/bin",
"LD_LIBRARY_PATH": "$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64",
"XLA_FLAGS": "--xla_gpu_cuda_data_dir=/usr/local/cuda"
},
"updateContentCommand": "bash .devcontainer/install-dev-tools.sh",
// "postCreateCommand": [
// "nvidia-smi"
// ],
"customizations" : {
"jetbrains" : {
"backend" : "CLion"
}
},
// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
// "remoteUser": "root"
}

0
backends/trtllm/.devcontainer/install-dev-tools.sh → .devcontainer/install-dev-tools.sh
View File

0
backends/trtllm/.devcontainer/reinstall-cmake.sh → .devcontainer/reinstall-cmake.sh
View File

26
backends/trtllm/.devcontainer/devcontainer.json
View File

@ -1,26 +0,0 @@
// For format details, see https://aka.ms/devcontainer.json. For config options, see the
// README at: https://github.com/devcontainers/templates/tree/main/src/cpp
{
"name": "CUDA",
"image": "nvidia/cuda:12.6.0-devel-ubuntu24.04",
// "runArgs": [
// "--gpus=all"
// ],
"remoteEnv": {
"PATH": "${containerEnv:PATH}:/usr/local/cuda/bin",
"LD_LIBRARY_PATH": "$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64",
"XLA_FLAGS": "--xla_gpu_cuda_data_dir=/usr/local/cuda"
},
"updateContentCommand": "bash backends/trtllm/.devcontainer/install-dev-tools.sh",
// "postCreateCommand": [
// "nvidia-smi"
// ],
"customizations" : {
"jetbrains" : {
"backend" : "CLion"
}
},
// Uncomment to connect as root instead. More info: https://aka.ms/dev-containers-non-root.
// "remoteUser": "root"
}