Fix user

Dockerfile

@@ -10,18 +10,6 @@ RUN apt-get update && apt-get install -y \
     	ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
 		&& rm -rf /var/lib/apt/lists/*
 
-# RUN apt-get -y update
-# RUN apt install -y software-properties-common
-# RUN add-apt-repository ppa:deadsnakes/ppa
-# RUN apt-get -y install python3.8
-# RUN apt-get -y install python3-pip
-# RUN apt install -y python3.8-distutils
-# RUN apt-get install -y gcc
-# RUN apt-get install -y python3.8-dev
-
-# RUN useradd -ms /bin/bash admin
-# USER admin
-
 RUN useradd -ms /bin/bash user
 USER user
 
@@ -35,25 +23,15 @@ RUN pyenv install 3.8.15 && \
     pyenv rehash && \
     pip install --no-cache-dir --upgrade pip setuptools wheel
 
+ENV FORCE_CUDA="1"
 ENV WORKDIR=/code
 WORKDIR $WORKDIR
 RUN chown -R user:user $WORKDIR
 RUN chmod 755 $WORKDIR
 
-# RUN nvidia-smi
-
 
 COPY requirements.txt $WORKDIR/requirements.txt
 COPY oneformer $WORKDIR/oneformer
-# RUN python3.8 --version
-# RUN which python3.8
-# RUN python3.8 -m pip install --upgrade pip
-# RUN python3.8 -m pip install multidict
-# RUN python3.8 -m pip install typing-extensions
-# RUN python3.8 -m pip install --upgrade setuptools
-# RUN python3.8 -m pip install wheel
-# RUN python3.8 -m pip install cython
-# RUN python3.8 -m pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
 
 COPY . .
@@ -63,8 +41,8 @@ RUN ls
 
 RUN sh deform_setup.sh
 
-USER user
+# USER user
 
-EXPOSE 7860
+# EXPOSE 7860
 
-ENTRYPOINT ["python", "gradio_app.py"]
+# ENTRYPOINT ["python", "gradio_app.py"]

oneformer/modeling/pixel_decoder/ops/make.sh

@@ -10,4 +10,4 @@
 # Copyright (c) Facebook, Inc. and its affiliates.
 # Modified by Bowen Cheng from https://github.com/fundamentalvision/Deformable-DETR
 
-FORCE_CUDA=1 python3 setup.py build install
+FORCE_CUDA=1 python setup.py build install