Newer
Older
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 pyproject.toml README.md ./
# Install DAN as a package
RUN pip install . --no-cache-dir