diff --git a/Dockerfile b/Dockerfile index 3bb6fff..eaf50a2 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,4 +1,6 @@ -ARG basetag="418.87.01-ubuntu18.04" # Ubuntu only +# Ubuntu only +ARG basetag="418.87.01-ubuntu18.04" + FROM nvidia/driver:"${basetag}" ENV DEBIAN_FRONTEND=noninteractive @@ -19,6 +21,11 @@ RUN apt-get update && \ ENV LC_ALL=C.UTF-8 RUN update-locale LC_ALL="C.UTF-8" +# Install dependencies +RUN python3 -m pip install --upgrade pip +COPY ./requirements.txt /tmp/requirements.txt +RUN pip3 install -r /tmp/requirements.txt + # Install nvitop COPY . /nvitop WORKDIR /nvitop