Dockerfile

# As of November 21, 2023, the lastest version of python that is supported by pytorch is 3.11.6
FROM python:3.11.6-bookworm

# Prevents the user from being prompted for input, thus preventing the build from hanging
ARG DEBIAN_FRONTEND=noninteractive

# Prevents python from writing pyc files to the container "disk"
ENV PYTHONDONTWRITEBYTECODE=1

# Prevents python from buffering stdin and stdout
ENV PYTHONUNBUFFERED=1

# Metadata
LABEL name="GraphComm-py3.11.6"
LABEL author="samueljcatania"
LABEL description="Dockerfile for GraphComm, running in python3.11.6"
LABEL version="1.0"

# Upgrade pip, if applicable
RUN pip3 install --upgrade pip

# As torch-cluster depends on pytorch, first install pytorch separately
RUN pip3 install --upgrade --no-cache-dir torch

# Install the remaining dependencies
COPY requirements.txt .
RUN pip3 install --upgrade-strategy only-if-needed --no-cache-dir --requirement requirements.txt

# Create a non-root user to use within the container
RUN addgroup --system graphcomm && adduser --system --group graphcomm
USER graphcomm

# Copy the source code into the container
COPY src /home/graphcomm/src

# Set the working directory in the container
WORKDIR /home/graphcomm