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

# Install DAN as a package
COPY dan dan
COPY requirements.txt *-requirements.txt setup.py VERSION README.md ./
RUN pip install . --no-cache-dir