praeclarumjj3 commited on
Commit
2685271
1 Parent(s): be0b089
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -35,11 +35,11 @@ RUN pip install --no-cache-dir --upgrade -r $WORKDIR/requirements.txt
35
 
36
  COPY . .
37
 
38
- RUN sh deform_setup.sh
39
-
40
  ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
41
 
42
- RUN ln -s ./oneformer/modeling/pixel_decoder/ops/ ./ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR install --user && cd ..
 
 
43
 
44
  # USER user
45
 
 
35
 
36
  COPY . .
37
 
 
 
38
  ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
39
 
40
+ RUN ln -s ./oneformer/modeling/pixel_decoder/ops/ ./ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR/oneformer/ install --user && cd ..
41
+
42
+ RUN sh deform_setup.sh
43
 
44
  # USER user
45