praeclarumjj3 commited on
Commit
b360236
·
1 Parent(s): 311617e
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -39,7 +39,7 @@ ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
39
 
40
  USER user
41
  RUN pip install ninja
42
- RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/ easy_install && cd ..
43
 
44
  RUN sh deform_setup.sh
45
 
 
39
 
40
  USER user
41
  RUN pip install ninja
42
+ 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 ..
43
 
44
  RUN sh deform_setup.sh
45