praeclarumjj3 commited on
Commit
f3fbc65
·
1 Parent(s): 0ffe942
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -38,7 +38,7 @@ COPY . .
38
 
39
  RUN sh deform_setup.sh
40
 
41
- RUN ln -s ./oneformer/modeling/pixel_decoder/ops/ ./ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build --build-base=$WORKDIR install --user && cd ..
42
 
43
  # USER user
44
 
 
38
 
39
  RUN sh deform_setup.sh
40
 
41
+ RUN ln -s ./oneformer/modeling/pixel_decoder/ops/ ./ && ls && cd ops/ && FORCE_CUDA=1 python TORCH_CUDA_ARCH_LIST="7.5+PTX" setup.py build --build-base=$WORKDIR install --user && cd ..
42
 
43
  # USER user
44