Skip to content

Commit

Permalink
Update base versions (#944)
Browse files Browse the repository at this point in the history
  • Loading branch information
mc-nv committed Nov 22, 2024
1 parent 5063a99 commit 5fa372c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ ARG BASE_IMAGE
ARG TRITONSDK_BASE_IMAGE

# DCGM version to install for Model Analyzer
ENV DCGM_VERSION=3.2.6
ENV DCGM_VERSION=3.3.6

# Ensure apt-get won't prompt for selecting options
ENV DEBIAN_FRONTEND=noninteractive
Expand All @@ -41,7 +41,7 @@ RUN mkdir -p /opt/triton-model-analyzer

RUN [ "$(uname -m)" != "x86_64" ] && arch="sbsa" || arch="x86_64" && \
curl -o /tmp/cuda-keyring.deb \
https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/$arch/cuda-keyring_1.0-1_all.deb && \
https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/$arch/cuda-keyring_1.1-1_all.deb && \
apt-get install /tmp/cuda-keyring.deb && rm /tmp/cuda-keyring.deb && \
apt-get update && apt-get install -y --no-install-recommends software-properties-common && \
apt-get install -y datacenter-gpu-manager=1:${DCGM_VERSION};
Expand Down

0 comments on commit 5fa372c

Please sign in to comment.