FROM ubuntu:18.04
LABEL io.k8s.display-name="NVIDIA DCGM Exporter"

COPY --from=builder /go/src/github.com/NVIDIA/gpu-monitoring-tools/dcgm-exporter /usr/bin/
COPY etc/dcgm-exporter /etc/dcgm-exporter

ARG DCGM_VERSION
RUN apt-get update && apt-get install -y --no-install-recommends \
    libgomp1 \
    wget && \
    rm -rf /var/lib/apt/lists/* && \
    wget --no-check-certificate https://developer.download.nvidia.com/compute/redist/dcgm/${DCGM_VERSION}/DEBS/datacenter-gpu-manager_${DCGM_VERSION}_amd64.deb && \
    dpkg -i datacenter-gpu-manager_*.deb && \
    rm -f datacenter-gpu-manager_*.deb

ENV NVIDIA_VISIBLE_DEVICES=all

RUN useradd dcgm-exporter
USER dcgm-exporter

ENTRYPOINT ["/usr/bin/dcgm-exporter"]

me@sshct:~$ /tmp/dcgmlibs/dcgm-exporter
INFO[0000] Starting dcgm-exporter
FATA[0000] libdcgm.so.1 not Found
me@sshct:~$ LD_LIBRARY_PATH=/tmp/dcgmlibs/ /tmp/dcgmlibs/dcgm-exporter
INFO[0000] Starting dcgm-exporter
Error: Failed to initialize NVML
FATA[0000] Error starting nv-hostengine: DCGM initialization error