Yuliang commited on
Commit
2a5bdc2
1 Parent(s): 337d241

update docker

Browse files
Files changed (2) hide show
  1. Dockerfile +13 -3
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -9,14 +9,20 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
9
  ffmpeg \
10
  libsm6 \
11
  libxext6 \
12
- libgl1-mesa-dri \
13
- libegl1-mesa \
 
 
 
 
14
  libgbm1 \
15
  build-essential \
16
  libeigen3-dev \
17
  python3.8 \
18
  python3-pip \
19
- nvidia-cuda-toolkit
 
 
20
 
21
  # Set up a new user named "user" with user ID 1000
22
  RUN useradd -m -u 1000 user
@@ -62,5 +68,9 @@ WORKDIR $HOME/app
62
 
63
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
64
  COPY --chown=user . $HOME/app
 
 
 
 
65
 
66
  CMD ["python", "app.py"]
 
9
  ffmpeg \
10
  libsm6 \
11
  libxext6 \
12
+ libfontconfig1 \
13
+ libegl1-mesa-dev \
14
+ libgl1-mesa-dev \
15
+ libgles2-mesa-dev \
16
+ libglib2.0-0 \
17
+ libxrender1 \
18
  libgbm1 \
19
  build-essential \
20
  libeigen3-dev \
21
  python3.8 \
22
  python3-pip \
23
+ nvidia-cuda-toolkit \
24
+ opencv-python-headless \
25
+ && rm -rf /var/lib/apt/lists/*
26
 
27
  # Set up a new user named "user" with user ID 1000
28
  RUN useradd -m -u 1000 user
 
68
 
69
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
70
  COPY --chown=user . $HOME/app
71
+ RUN chmod 755 $HOME/app
72
+
73
+ ENV TRANSFORMERS_CACHE=/tmp
74
+ ENV MPLCONFIGDIR=/tmp
75
 
76
  CMD ["python", "app.py"]
requirements.txt CHANGED
@@ -13,7 +13,7 @@ rtree
13
  pytorch_lightning
14
  kornia>0.4.0
15
  chumpy
16
- opencv-python
17
  opencv_contrib_python
18
  scikit-learn
19
  protobuf
 
13
  pytorch_lightning
14
  kornia>0.4.0
15
  chumpy
16
+ opencv-python-headless==4.8.1.78
17
  opencv_contrib_python
18
  scikit-learn
19
  protobuf