Hugging Face's logo

Spaces:
shi-labs
/
OneFormer
Running on T4

App Files Community
9
praeclarumjj3 commited on
Commit
be0b089
•
1 Parent(s): bc4e04e

Fix user

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -23,7 +23,6 @@ RUN pyenv install 3.8.15 && \
23
  pyenv rehash && \
24
  pip install --no-cache-dir --upgrade pip setuptools wheel
25
 
26
- ENV FORCE_CUDA="1"
27
  ENV WORKDIR=/code
28
  WORKDIR $WORKDIR
29
  RUN chown -R user:user $WORKDIR
@@ -38,7 +37,9 @@ 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 TORCH_CUDA_ARCH_LIST=7.5+PTX setup.py build --build-base=$WORKDIR install --user && cd ..
 
 
42
 
43
  # USER user
44
 
 
23
  pyenv rehash && \
24
  pip install --no-cache-dir --upgrade pip setuptools wheel
25
 
 
26
  ENV WORKDIR=/code
27
  WORKDIR $WORKDIR
28
  RUN chown -R user:user $WORKDIR
 
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