praeclarumjj3 commited on
Commit
0dd3195
1 Parent(s): 14a3656
Files changed (1) hide show
  1. Dockerfile +4 -3
Dockerfile CHANGED
@@ -37,10 +37,11 @@ COPY . .
37
 
38
  ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
39
 
40
- USER user
41
  RUN pip install ninja
42
- RUN chown -R user:user $HOME
43
- RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ install --user && cd ..
 
44
 
45
  RUN sh deform_setup.sh
46
 
 
37
 
38
  ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
39
 
40
+
41
  RUN pip install ninja
42
+
43
+ USER root
44
+ RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && pip install -e . && cd ..
45
 
46
  RUN sh deform_setup.sh
47