# Use the official Python image as the base image
FROM nvidia/cuda:11.0.3-base-ubuntu20.04

# Run update
RUN apt update

RUN apt-get install -y software-properties-common
RUN add-apt-repository ppa:deadsnakes/ppa
RUN apt-get update && apt-get install ffmpeg libsm6 libxext6  -y

#Install Python

RUN apt-get install -y python3.9 
RUN apt-get install -y python3-pip


# Set the working directory inside the container
WORKDIR /app


# Copy the contents of the cloned GitHub repository to the working directory in the container
COPY ./ /app

# Install Python dependencies
RUN python3.9 -m pip install numpy==1.21.6
RUN python3.9 -m pip install -r requirements.txt
RUN python3.9 -m pip install scikit-learn==0.24.2
ENV DEBIAN_FRONTEND=noninteractive
RUN apt update && apt install -y build-essential && rm -rf /var/lib/apt/lists/*
RUN python3.9 -m pip install scikit-image

#Give permission to script
RUN chmod +x ./entrypoint.sh

# Set the environment variable for GPU support
ENV NVIDIA_VISIBLE_DEVICES all

# Run the Python program when the container starts
ENTRYPOINT ["./entrypoint.sh"]