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

RUN apt-get update && apt-get install -y \
    python3.10 \
    python3-pip \
    python3-dev \
    libpq-dev \
    build-essential \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /workspace

# Upgrade pip to the latest version to fix the backtracking freeze!
RUN pip3 install --upgrade pip

COPY requirements.txt .
RUN pip3 install --default-timeout=1000 --no-cache-dir -r requirements.txt

RUN ln -s /usr/bin/python3 /usr/bin/python