mirror of
https://github.com/XuehaiPan/nvitop.git
synced 2026-05-15 06:06:12 -06:00
38 lines
1.3 KiB
Docker
38 lines
1.3 KiB
Docker
ARG basetag="450-signed-ubuntu22.04" # Ubuntu only
|
|
FROM nvcr.io/nvidia/driver:"${basetag}"
|
|
|
|
ENV NVIDIA_DISABLE_REQUIRE=true
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
|
|
# Update APT sources
|
|
RUN . /etc/os-release && [ "${NAME}" = "Ubuntu" ] && \
|
|
echo "deb [arch=amd64] http://archive.ubuntu.com/ubuntu ${UBUNTU_CODENAME} main universe" > /etc/apt/sources.list && \
|
|
echo "deb [arch=amd64] http://archive.ubuntu.com/ubuntu ${UBUNTU_CODENAME}-updates main universe" >> /etc/apt/sources.list && \
|
|
echo "deb [arch=amd64] http://archive.ubuntu.com/ubuntu ${UBUNTU_CODENAME}-security main universe" >> /etc/apt/sources.list
|
|
|
|
# Install Python 3
|
|
RUN apt-get update && \
|
|
apt-get install --quiet --yes --no-install-recommends python3-dev python3-venv locales && \
|
|
rm -rf /var/lib/apt/lists/*
|
|
|
|
# Setup locale
|
|
ENV LC_ALL=C.UTF-8
|
|
RUN update-locale LC_ALL="C.UTF-8"
|
|
|
|
# Setup environment
|
|
RUN python3 -m venv /venv && \
|
|
. /venv/bin/activate && \
|
|
python3 -m pip install --upgrade pip setuptools && \
|
|
rm -rf /root/.cache && \
|
|
echo && echo && echo "source /venv/bin/activate" >> /root/.bashrc
|
|
ENV SHELL /bin/bash
|
|
|
|
# Install nvitop
|
|
COPY . /nvitop
|
|
WORKDIR /nvitop
|
|
RUN . /venv/bin/activate && \
|
|
python3 -m pip install . && \
|
|
rm -rf /root/.cache
|
|
|
|
# Entrypoint
|
|
ENTRYPOINT [ "/venv/bin/python3", "-m", "nvitop" ]
|