Dockerfile

# BUILD: docker build . -t nvtop
# or use other image with: --build-arg IMAGE=nvcr.io/nvidia/cudagl:11.4.2-base-ubuntu20.04
# or nvidia/driver:418.87.01-ubuntu18.04, nvcr.io/nvidia/cudagl:11.4.2-base-ubuntu20.04
# USE: docker run --rm -it --gpus all --pid host nvtop

ARG IMAGE=nvidia/opengl:1.2-glvnd-runtime-ubuntu18.04

FROM ${IMAGE} as builder

ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get update && \
  apt-get install -yq build-essential wget libncurses5-dev libncursesw5-dev libssl-dev \
  pkg-config libdrm-dev libgtest-dev libudev-dev

RUN wget https://github.com/Kitware/CMake/releases/download/v3.18.0/cmake-3.18.0.tar.gz
RUN tar zxf cmake-3.18.0.tar.gz
RUN ./cmake-3.18.0/bootstrap --prefix=/usr && make && make install

COPY . /nvtop
WORKDIR /nvtop
RUN mkdir -p /nvtop/build && \
  cd /nvtop/build && \
  cmake .. && \
  make -j && \
  make install

# Stage 2
FROM ${IMAGE}
RUN DEBIAN_FRONTEND=noninteractive apt-get update -y && apt-get install -yq libncurses5 libncursesw5 libdrm-amdgpu1 \
  && rm -rf /var/lib/apt/lists/*
COPY --from=builder /usr/local/bin/nvtop /usr/local/bin/nvtop
COPY --from=builder /usr/local/share/man/man1/nvtop.1 /usr/local/share/man/man1/nvtop.1

ENV LANG=C.UTF-8

ENTRYPOINT [ "/usr/local/bin/nvtop" ]