content: # Note: as of April 2025, this Dockerfile is outdated and requires adjustments
  
  # Inspiration for setup @ https://dev.to/ordigital/nvidia-525-cuda-118-python-310-pytorch-gpu-docker-image-1l4a
  FROM docker.io/nvidia/cuda:11.8.0-devel-ubuntu22.04
  
  ENV PYTHONUNBUFFERED=1
  
  # SYSTEM
  RUN apt-get update --yes --quiet && DEBIAN_FRONTEND=noninteractive apt-get install --yes --quiet --no-install-recommends \
      software-properties-common \
      build-essential apt-utils \
      wget curl vim git ca-certificates kmod \
      nvidia-driver-525 \
   && rm -rf /var/lib/apt/lists/*
  
  # PYTHON 3.10
  RUN add-apt-repository --yes ppa:deadsnakes/ppa && apt-get update --yes --quiet
  RUN DEBIAN_FRONTEND=noninteractive apt-get install --yes --quiet --no-install-recommends \
      python3.10 \
      python3.10-dev \
      python3.10-distutils \
      python3.10-lib2to3 \
      python3.10-gdbm \
      python3.10-tk \
      pip
  
  RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 999 \
      && update-alternatives --config python3 && ln -s /usr/bin/python3 /usr/bin/python
  
  RUN pip install --upgrade pip
  
  # Create and set the working directory
  RUN mkdir -p /OneTrainer
  WORKDIR /OneTrainer
  
  # Copy the current directory's contents to the container image
  COPY . /OneTrainer
  WORKDIR /OneTrainer
  
  # Install requirements
  RUN python3 --version
  RUN python3 -m pip install -r requirements.txt
  
  # Run the training UI
  CMD ["python3", "scripts/train_ui.py"]
  