Yuliang commited on
Commit
db57386
1 Parent(s): a6610ba

update dev mode

Browse files
Files changed (2) hide show
  1. Dockerfile +4 -5
  2. requirements.txt +1 -1
Dockerfile CHANGED
@@ -25,7 +25,6 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
25
  python3-pip \
26
  python-is-python3 \
27
  nvidia-cuda-toolkit \
28
- opencv-python-headless \
29
  && rm -rf /var/lib/apt/lists/*
30
 
31
  # Set up a new user named "user" with user ID 1000
@@ -49,7 +48,7 @@ ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH}
49
 
50
 
51
  # Set home to the user's home directory
52
- ENV HOME=/home/user \
53
  PATH=/home/user/.local/bin:$PATH \
54
  PYTHONPATH=$HOME/app \
55
  PYTHONUNBUFFERED=1 \
@@ -72,10 +71,10 @@ RUN chmod 777 $HOME
72
  ENV TRANSFORMERS_CACHE=/tmp
73
  ENV MPLCONFIGDIR=/tmp
74
 
75
- # Set the working directory to the user's home directory
76
- WORKDIR $HOME/app
77
-
78
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
79
  COPY --chown=user . $HOME/app
80
 
 
 
 
81
  CMD ["python", "app.py"]
 
25
  python3-pip \
26
  python-is-python3 \
27
  nvidia-cuda-toolkit \
 
28
  && rm -rf /var/lib/apt/lists/*
29
 
30
  # Set up a new user named "user" with user ID 1000
 
48
 
49
 
50
  # Set home to the user's home directory
51
+ ENV HOME=/home/user/app \
52
  PATH=/home/user/.local/bin:$PATH \
53
  PYTHONPATH=$HOME/app \
54
  PYTHONUNBUFFERED=1 \
 
71
  ENV TRANSFORMERS_CACHE=/tmp
72
  ENV MPLCONFIGDIR=/tmp
73
 
 
 
 
74
  # Copy the current directory contents into the container at $HOME/app setting the owner to the user
75
  COPY --chown=user . $HOME/app
76
 
77
+ # Set the working directory to the user's home directory
78
+ WORKDIR $HOME/app
79
+
80
  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-headless==4.8.1.78
17
  opencv_contrib_python
18
  scikit-learn
19
  protobuf
 
13
  pytorch_lightning
14
  kornia>0.4.0
15
  chumpy
16
+ opencv-python
17
  opencv_contrib_python
18
  scikit-learn
19
  protobuf