From 8e25428713bdf26de8f9e2812c7ef4112d952ae8 Mon Sep 17 00:00:00 2001 From: Morgan Funtowicz Date: Fri, 28 Jun 2024 15:16:10 +0200 Subject: [PATCH] Update devcontainer to remove clang and base image on PyTorch --- .devcontainer/devcontainer.json | 2 +- .devcontainer/install-dev-tools.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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