update import detectron2

Dockerfile

@@ -1,4 +1,4 @@
-# Base image with Python 3.9 for app dependencies
+# Stage 1: Base Python image
 FROM python:3.9 as base
 
 # Create a user and set environment variables
@@ -9,23 +9,20 @@ ENV PATH="/home/user/.local/bin:$PATH"
 # Set working directory
 WORKDIR /app
 
-# Copy and install Python dependencies
+# Copy Python dependencies and install them
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
 # Copy application files
 COPY --chown=user . /app
 
-# Expose application port and set default command
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
-
-# Base image for CUDA support
+# Stage 2: CUDA and Detectron2
 FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04 as cuda
 
 # Use noninteractive mode for apt-get
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install dependencies
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     python3-opencv \
     python3-pip \
@@ -36,15 +33,16 @@ RUN apt-get update && apt-get install -y \
     sudo \
     ninja-build
 
-# Link Python and Pip to default commands
+# Link Python and Pip commands
 RUN ln -sv /usr/bin/python3 /usr/bin/python && ln -sv /usr/bin/pip3 /usr/bin/pip
 
-# Install Python packages
-RUN pip install --no-cache-dir --upgrade -r requirements.txt
-RUN pip install --user tensorboard cmake onnx
+# Copy dependencies from the base stage
+COPY --from=base /home/user/.local /home/user/.local
+COPY --from=base /app /app
 
-# Install fvcore
-RUN pip install --user 'git+https://github.com/facebookresearch/fvcore'
+# Install additional Python packages
+RUN pip install --no-cache-dir tensorboard cmake onnx
+RUN pip install --no-cache-dir 'git+https://github.com/facebookresearch/fvcore'
 
 # Clone Detectron2 repository
 RUN git clone https://github.com/facebookresearch/detectron2 detectron2_repo
@@ -53,12 +51,15 @@ RUN git clone https://github.com/facebookresearch/detectron2 detectron2_repo
 ENV FORCE_CUDA="1"
 ARG TORCH_CUDA_ARCH_LIST="Kepler;Kepler+Tesla;Maxwell;Maxwell+Tegra;Pascal;Volta;Turing"
 ENV TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST}"
+ENV FVCORE_CACHE="/tmp"
 
 # Install Detectron2 in editable mode
-RUN pip install --user -e detectron2_repo
+RUN pip install --no-cache-dir -e detectron2_repo
 
-# Set fvcore cache directory
-ENV FVCORE_CACHE="/tmp"
+# Set the working directory and environment
+USER user
+ENV HOME=/home/user
+WORKDIR $HOME/app
 
-# Set working directory
-WORKDIR /home/user/detectron2_repo
+# Expose application port and set default command
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]