FROM nvidia/cuda:12.2.0-base-ubuntu22.04

# Install python and pip
RUN apt-get -y update && \
    apt-get -y install python3 python3-pip && \
    apt-get clean -y

WORKDIR /src

# Copy DAN data
COPY dan dan
COPY pyproject.toml README.md ./

# Install DAN as a package
RUN pip install . --no-cache-dir