Dockerfile

FROM nvidia/cuda:11.4.3-cudnn8-devel-ubuntu20.04

ARG DEBIAN_FRONTEND=noninteractive

# Keeps Python from generating .pyc files in the container
ENV PYTHONDONTWRITEBYTECODE=1

# Turns off buffering for easier container logging
ENV PYTHONUNBUFFERED=1

# Install Python 3.9
RUN apt update && \
    apt install -y software-properties-common && \
    add-apt-repository ppa:deadsnakes/ppa && \
    apt update && \
    apt install -y python3.9 python3-pip python3.9-distutils git && \
    update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1 && \
    update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1 && \
    update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1

# OpenGL is needed for OpenCV
RUN apt update && \
    apt install -y libgl1-mesa-glx vim

# Install poetry
# RUN pip install --upgrade pip setuptools && \
#     pip install poetry && \
#     poetry config virtualenvs.create true && \
#     poetry config virtualenvs.in-project true

# Install venv
RUN apt install -y python3.9-venv python3.9-dev && \
    python -m venv /mil_env && \
    . /mil_env/bin/activate && \
    pip install --upgrade pip && \
    pip install wheel numpy && \
    pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 --extra-index-url https://download.pytorch.org/whl/cu113 && \
    pip install pyg_lib==0.2.0 torch_cluster==1.6.0 torch_scatter==2.1.0 torch_sparse==0.6.16 torch_spline_conv==1.2.1 -f https://data.pyg.org/whl/torch-1.12.1+cu113.html && \
    pip install torch_geometric

# Install dependencies
ADD requirements.txt /tmp/requirements.txt
RUN . /mil_env/bin/activate && pip install -r /tmp/requirements.txt

# Install dependencies
# RUN mkdir -p /app
# WORKDIR /app
# COPY ./pyproject.toml ./poetry.lock* ./
# RUN poetry install --no-root
# RUN rm -rf /app

VOLUME /app
VOLUME /data

ENV PYTHONPATH "/app:${PYTHONPATH}"

ENV DATA_FOLDER /data
ENV HYDRA_FULL_ERROR 1

# Creates a non-root user with an explicit UID and adds permission to access the /app folder
# For more info, please refer to https://aka.ms/vscode-docker-python-configure-containers
# RUN mkdir -p /app
# RUN adduser -u 5678 --disabled-password --gecos "" appuser && chown -R appuser /app
# USER appuser
WORKDIR /app

RUN git config --global --add safe.directory /app

# During debugging, this entry point will be overridden. For more information, please refer to https://aka.ms/vscode-docker-python-debug
CMD sleep 365d