diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json index 2f33406c..4402349a 100644 --- a/.devcontainer/devcontainer.json +++ b/.devcontainer/devcontainer.json @@ -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" ], diff --git a/.devcontainer/install-dev-tools.sh b/.devcontainer/install-dev-tools.sh index bde5828c..2dbd0800 100644 --- a/.devcontainer/install-dev-tools.sh +++ b/.devcontainer/install-dev-tools.sh @@ -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