fleet-intelligence-agent/Dockerfile at main · NVIDIA/fleet-intelligence-agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
ARG DCGM_VERSION="4.4.2-1-ubuntu22.04"

FROM golang:1.26.1 AS build

RUN apt-get update && apt-get install -y --no-install-recommends \
    ca-certificates \
    git \
    build-essential \
  && rm -rf /var/lib/apt/lists/*

WORKDIR /src

COPY go.mod go.sum ./
# Local replace target must exist before go mod download (replace => ./third_party/fleet-intelligence-sdk).
COPY third_party/fleet-intelligence-sdk ./third_party/fleet-intelligence-sdk

RUN go mod download

COPY . .

ARG TARGETOS=linux
ARG TARGETARCH
ARG BUILD_TIMESTAMP=""
ARG VERSION="0.0.1+container"
ARG REVISION=""

RUN CGO_ENABLED=1 GOOS=${TARGETOS} GOARCH=${TARGETARCH:-amd64} \
  go build -trimpath \
    -ldflags "-s -w \
      -X github.com/NVIDIA/fleet-intelligence-agent/internal/version.BuildTimestamp=${BUILD_TIMESTAMP} \
      -X github.com/NVIDIA/fleet-intelligence-agent/internal/version.Version=${VERSION} \
      -X github.com/NVIDIA/fleet-intelligence-agent/internal/version.Revision=${REVISION} \
      -X github.com/NVIDIA/fleet-intelligence-agent/internal/version.Package=github.com/NVIDIA/fleet-intelligence-agent" \
    -o /out/fleetint ./cmd/fleetint

FROM nvidia/dcgm:${DCGM_VERSION} AS runtime

ENV DEBIAN_FRONTEND=noninteractive
RUN set -eux; \
  apt-get update && apt-get install -y --no-install-recommends \
    ca-certificates \
    bash \
    sudo \
    kmod \
    util-linux \
    dmidecode \
    pciutils \
    wget \
    gnupg \
    gnupg2 \
  ; \
  arch="$(dpkg --print-architecture)"; \
  case "${arch}" in \
    amd64) cuda_arch="x86_64" ;; \
    arm64) cuda_arch="sbsa" ;; \
    *) echo "unsupported arch: ${arch}" >&2; exit 1 ;; \
  esac; \
  wget -qO /usr/share/keyrings/cuda-archive-keyring.gpg \
    "https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${cuda_arch}/cuda-archive-keyring.gpg"; \
  echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] \
    https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/${cuda_arch}/ /" \
    > /etc/apt/sources.list.d/cuda.list; \
  apt-get update; \
  apt-get install -y --no-install-recommends \
    nvattest \
    corelib \
  ; \
  rm -rf /var/lib/apt/lists/*

COPY --from=build /out/fleetint /usr/bin/fleetint

EXPOSE 15133
VOLUME ["/var/lib/fleetint"]

ENTRYPOINT ["/usr/bin/fleetint"]
CMD ["run"]