Hugging Face's logo

Spaces:
shi-labs
/
OneFormer
Running on T4

App Files Community
9
praeclarumjj3 commited on
Commit
23144ba
•
1 Parent(s): b715310

Root user

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
2
- CMD nvidia-smi
3
 
4
  ENV DEBIAN_FRONTEND noninteractive
5
  RUN apt-get update && apt-get install -y \
@@ -41,7 +41,7 @@ ARG TORCH_CUDA_ARCH_LIST=7.5+PTX
41
  RUN pip install ninja
42
 
43
  USER user
44
- RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 python setup.py build install --user && cd ..
45
  RUN sh deform_setup.sh
46
 
47
  USER user
 
1
  FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
2
+ CMD sh deform_setup.sh
3
 
4
  ENV DEBIAN_FRONTEND noninteractive
5
  RUN apt-get update && apt-get install -y \
 
41
  RUN pip install ninja
42
 
43
  USER user
44
+ RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd ops/ && FORCE_CUDA=1 pip install --user --ignore-installed . && cd ..
45
  RUN sh deform_setup.sh
46
 
47
  USER user