From fdc98654cc9b2c7de0ccdfb3cfad2a73a6699d74 Mon Sep 17 00:00:00 2001 From: Iman Tabrizian Date: Wed, 10 Apr 2024 10:48:31 -0400 Subject: [PATCH] Review comment --- .devcontainer/Dockerfile | 2 +- .devcontainer/devcontainer.json | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/.devcontainer/Dockerfile b/.devcontainer/Dockerfile index b67a472e..a7aa2b9c 100644 --- a/.devcontainer/Dockerfile +++ b/.devcontainer/Dockerfile @@ -24,7 +24,7 @@ # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. -FROM gitlab-master.nvidia.com:5005/dl/dgx/tritonserver:master-py3-devel +FROM tritonserver_devel ARG USERNAME=triton-server diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json index 8e6385ac..9dfa72f1 100644 --- a/.devcontainer/devcontainer.json +++ b/.devcontainer/devcontainer.json @@ -19,7 +19,6 @@ "runArgs": [ "--cap-add=SYS_PTRACE", "--security-opt", "seccomp=unconfined", "--gpus=all", "--shm-size=2g", "--ulimit", "stack=67108864"], "mounts": [ "source=${localEnv:HOME}/.ssh,target=/home/triton-server/.ssh,type=bind,consistency=cached", - "source=${localEnv:HOME}/scratch,target=/scratch,type=bind,consistency=cached", "source=${localEnv:HOME}/.cache/huggingface,target=/home/triton-server/.cache/huggingface,type=bind,consistency=cached" ] }