# Use the specified CUDA image as the base image
FROM nvidia/cuda:12.0.0-cudnn8-devel-ubuntu20.04

# Set environment variables to avoid interactive prompts during package installation
ENV DEBIAN_FRONTEND=noninteractive

# Change the APT package server (fix "Hash Sum Mismatch" problem)
RUN sed -i 's/archive.ubuntu.com/repo.catswords.com/g' /etc/apt/sources.list
RUN sed -i 's/security.ubuntu.com/repo.catswords.com/g' /etc/apt/sources.list

# Update the package list and install Python 3 and necessary packages
RUN apt-get update && \
    apt-get install -y python3 python3-pip python3-dev && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# Set Python 3 as the default python
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3 1

# Create a working directory
WORKDIR /app

# Copy the requirements.txt file and install Python packages if available
COPY requirements.txt ./
RUN pip3 install --no-cache-dir -r requirements.txt

# Command to run the application (update this as needed)
CMD ["tail", "-f", "/dev/null"]