FROM nvidia/cuda:12.3.2-base-ubuntu22.04
LABEL authors="zxasv"

# Set environment variables
ENV DEBIAN_FRONTEND=noninteractive

# Install system dependencies
RUN apt-get update && \
    apt-get install -y \
        git \
        python3-pip \
        python3-dev \
        python3-opencv \
        libglib2.0-0
# Install any python packages you need
COPY requirements.txt requirements.txt
RUN ls -la /
RUN python3 -m pip install --upgrade pip
RUN python3 -m pip install --no-cache-dir -r requirements.txt
# Upgrade pip
RUN python3 -m pip install --upgrade pip

# Install PyTorch and torchvision
RUN pip3 install torch torchvision torchaudio

# Set the working directory
WORKDIR /app

COPY / /
RUN ls -la /

# Set the entrypoint
ENTRYPOINT [ "python3" ]