Dockerfile

FROM nvcr.io/nvidia/cuda:11.8.0-base-ubuntu22.04
ENV DEBIAN_FRONTEND=noninteractive

# Ensure system packages are updated and install prerequisites
RUN apt-get update && apt-get install -y \
    software-properties-common \
    build-essential \
    wget \
    curl \
    libssl-dev \
    zlib1g-dev \
    libbz2-dev \
    libreadline-dev \
    libsqlite3-dev \
    libffi-dev \
    libncurses5-dev \
    libncursesw5-dev \
    xz-utils \
    tk-dev \
    liblzma-dev \
    git \
    ca-certificates \
    unzip \
    git-lfs \
    && rm -rf /var/lib/apt/lists/* \
    && git lfs install

# Install Python 3.9.20 from source
RUN wget https://www.python.org/ftp/python/3.9.20/Python-3.9.20.tgz \
    && tar -xf Python-3.9.20.tgz \
    && cd Python-3.9.20 \
    && ./configure --enable-optimizations \
    && make -j$(nproc) \
    && make altinstall \
    && cd .. \
    && rm -rf Python-3.9.20 Python-3.9.20.tgz

# Ensure pip is upgraded
RUN python3.9 -m pip install --upgrade pip

# Set Python 3.9 as the default python
RUN update-alternatives --install /usr/bin/python python /usr/local/bin/python3.9 1

# Optional: Set python3 and python alias to python3.9
RUN ln -sf /usr/local/bin/python3.9 /usr/bin/python3 \
    && ln -sf /usr/local/bin/python3.9 /usr/bin/python

RUN pip install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2

COPY ./requirements.txt /root

RUN cd /root \
    && pip install -r requirements.txt

WORKDIR /root/DirectDrag