Dockerfile

FROM nvidia/cuda:11.0.3-devel-ubuntu20.04

# Set lang
ENV LC_ALL C.UTF-8
ENV LANG C.UTF-8

# Set python env vars
ENV PYTHONUNBUFFERED=1 \
    # Prevents python creating .pyc files
    PYTHONDONTWRITEBYTECODE=1 \
    \
    # Pip
    PIP_NO_CACHE_DIR=1 \
    PIP_DISABLE_PIP_VERSION_CHECK=1 \
    \
    # Poetry
    # https://python-poetry.org/docs/configuration/#using-environment-variables
    POETRY_VERSION=1.1.11 \
    POETRY_NO_INTERACTION=1 \
    POETRY_HOME=/opt/poetry \
    PATH="/opt/poetry/bin:$PATH"

# Install python 3.8
ENV DEBIAN_FRONTEND=noninteractive

# Rotating Keys
RUN rm /etc/apt/sources.list.d/cuda.list

RUN apt-get update && \
    apt-get install -y --no-install-recommends \
    git \
    curl \
    git-lfs \
    ssh \
    python3.8 \
    python3.8-dev \
    python3.8-venv \
    python3-pip \
    && rm -rf /var/lib/apt/lists/*

# install poetry - respects $POETRY_VERSION & $POETRY_HOME
WORKDIR /opt
RUN curl -sSL https://raw.githubusercontent.com/python-poetry/poetry/master/install-poetry.py | python3.8 -

# install poetry dependencies
COPY pyproject.toml poetry.lock ./
RUN poetry config virtualenvs.create false && \
    poetry install

# Copy source code to the image
COPY --chown=$USER_UID:$USER_GID . ./simulation
ENV PYTHONPATH=/opt/simulation/src:$PYTHONPATH