Dockerfile

FROM nvcr.io/nvidia/pytorch:22.01-py3

# Install system dependencies
ENV TZ=Europe/Berlin \
    DEBIAN_FRONTEND=noninteractive

# Install required packages for adding repositories
RUN apt-get update && apt-get install -y \
    curl \
    build-essential \
    wget \
    gnupg \
    python3-setuptools \
    software-properties-common

# Add the PackageCloud repository for git-lfs
RUN curl -s https://packagecloud.io/install/repositories/github/git-lfs/script.deb.sh | bash

# Install git-lfs
RUN apt-get update && apt-get install -y git-lfs

# Initialize git-lfs
RUN git lfs install

# Clean up
RUN apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# Install Rust
RUN curl -sSf https://sh.rustup.rs | sh -s -- -y
ENV PATH="/root/.cargo/bin:${PATH}"

# Set working directory
WORKDIR /workspace

# Install core build dependencies
RUN python3 -m pip install --no-cache-dir --upgrade \
    pip \
    setuptools==68.0.0 \
    wheel \
    setuptools_scm


RUN pip install --upgrade pip
# Copy requirements first to leverage Docker cache
COPY requirements.txt .

# Install dependencies one by one to handle build issues
RUN while read requirement; do \
        echo "Installing $requirement..." && \
        pip install --no-cache-dir $requirement || pip install --no-cache-dir --no-deps $requirement; \
    done < requirements.txt


RUN pip install --upgrade packaging

# Copy project files
COPY . .

# Install package in development mode without dependencies
RUN pip install --no-deps -e .

# Clear cache
RUN apt-get clean && rm -rf /var/lib/apt/lists/*