praeclarumjj3 commited on
Commit
1bc5de5
1 Parent(s): f2f8acf
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -40,9 +40,11 @@ ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
40
 
41
  RUN pip install ninja
42
 
43
- USER user
44
  RUN chown -R user:user /usr/local
45
  RUN chmod -R 777 /usr/local
 
 
46
  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 ..
47
  RUN sh deform_setup.sh
48
 
 
40
 
41
  RUN pip install ninja
42
 
43
+ USER root
44
  RUN chown -R user:user /usr/local
45
  RUN chmod -R 777 /usr/local
46
+
47
+ USER user
48
  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 ..
49
  RUN sh deform_setup.sh
50