huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-05-04 16:52:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update devcontainer to remove clang and base image on PyTorch

Browse Source
This commit is contained in:
Morgan Funtowicz 2024-06-28 15:16:10 +02:00
parent f3e729a6d6
commit 8e25428713
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.devcontainer/devcontainer.json
View File

@ -1,6 +1,6 @@
{ {
"name": "tgi-trtllm-backend", "name": "tgi-trtllm-backend",
"image": "nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04", "image": "nvcr.io/nvidia/pytorch:24.05-py3",
"runArgs": [ "runArgs": [
"--gpus=all" "--gpus=all"
], ],

2
.devcontainer/install-dev-tools.sh
View File

@ -3,7 +3,7 @@ apt-get update
apt-get upgrade -y apt-get upgrade -y
# install Linux tools and Python 3 # install Linux tools and Python 3
apt-get install software-properties-common wget curl clang cmake git libopenmpi-dev libssl-dev \ apt-get install software-properties-common wget curl cmake gcc g++ git libopenmpi-dev libssl-dev nija-build \
python3-dev python3-pip python3-wheel python3-setuptools -y python3-dev python3-pip python3-wheel python3-setuptools -y
# install Python packages # install Python packages