Mirrors
/
hf_text-generation-inference
mirror of https://github.com/huggingface/text-generation-inference.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update devcontainer to remove clang and base image on PyTorch

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",
"image": "nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04",
"image": "nvcr.io/nvidia/pytorch:24.05-py3",
"runArgs": [
"--gpus=all"
],

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

@ -3,7 +3,7 @@ apt-get update
apt-get upgrade -y
# 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
# install Python packages