praeclarumjj3 commited on
Commit
5c23f00
1 Parent(s): 7f81961
Files changed (2) hide show
  1. Dockerfile +13 -7
  2. deform_setup.sh +0 -1
Dockerfile CHANGED
@@ -10,11 +10,16 @@ RUN apt-get update && apt-get install -y \
10
  ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
11
  && rm -rf /var/lib/apt/lists/*
12
 
13
- RUN useradd -ms /bin/bash user
14
- USER user
15
 
16
- ENV HOME=/home/user \
17
- PATH=/home/user/.local/bin:$PATH
 
 
 
 
 
18
 
19
  RUN curl https://pyenv.run | bash
20
  ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
@@ -25,7 +30,8 @@ RUN pyenv install 3.8.15 && \
25
 
26
  ENV WORKDIR=/code
27
  WORKDIR $WORKDIR
28
- RUN chown -R user:user $WORKDIR
 
29
  RUN chmod -R 755 $WORKDIR
30
 
31
 
@@ -45,10 +51,10 @@ RUN ln -s $WORKDIR/oneformer/modeling/pixel_decoder/ops/ $WORKDIR/ && ls && cd o
45
 
46
  RUN sh deform_setup.sh
47
 
48
- USER user
49
  RUN sh deform_setup.sh
50
 
51
- USER user
52
 
53
  EXPOSE 7860
54
 
 
10
  ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx \
11
  && rm -rf /var/lib/apt/lists/*
12
 
13
+ # RUN useradd -ms /bin/bash user
14
+ # USER user
15
 
16
+ USER root
17
+
18
+ # ENV HOME=/home/user \
19
+ # PATH=/home/user/.local/bin:$PATH
20
+
21
+ ENV HOME=/home/root \
22
+ PATH=/home/root/.local/bin:$PATH
23
 
24
  RUN curl https://pyenv.run | bash
25
  ENV PATH=$HOME/.pyenv/shims:$HOME/.pyenv/bin:$PATH
 
30
 
31
  ENV WORKDIR=/code
32
  WORKDIR $WORKDIR
33
+ # RUN chown -R user:user $WORKDIR
34
+ RUN chown -R root:root $WORKDIR
35
  RUN chmod -R 755 $WORKDIR
36
 
37
 
 
51
 
52
  RUN sh deform_setup.sh
53
 
54
+ USER root
55
  RUN sh deform_setup.sh
56
 
57
+ USER root
58
 
59
  EXPOSE 7860
60
 
deform_setup.sh CHANGED
@@ -3,7 +3,6 @@
3
  # ln -s ./oneformer/modeling/pixel_decoder/ops/ ./
4
  # ls
5
  # cd ops/ && bash make.sh && cd ..
6
- nvidia-smi
7
  echo '----------------------------------------------------------------'
8
  echo '----------------------------------------------------------------'
9
  pip3 freeze | grep MultiScaleDeformableAttention
 
3
  # ln -s ./oneformer/modeling/pixel_decoder/ops/ ./
4
  # ls
5
  # cd ops/ && bash make.sh && cd ..
 
6
  echo '----------------------------------------------------------------'
7
  echo '----------------------------------------------------------------'
8
  pip3 freeze | grep MultiScaleDeformableAttention