Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -4
Dockerfile
CHANGED
@@ -36,6 +36,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
36 |
python3.8-dev \
|
37 |
&& rm -rf /var/lib/apt/lists/*
|
38 |
|
|
|
39 |
# Set Python 3.8 as the default python and pip versions
|
40 |
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1
|
41 |
RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
|
@@ -57,13 +58,10 @@ RUN python basicsr/setup.py develop
|
|
57 |
# Install additional Python packages
|
58 |
RUN pip install dlib ffmpeg-python
|
59 |
|
60 |
-
# Initialize and update Git submodules (if not handled by the clone command)
|
61 |
-
RUN git submodule init && git submodule update
|
62 |
-
|
63 |
# Set the environment variable to specify the GPU device
|
64 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
65 |
ENV CUDA_VISIBLE_DEVICES=0
|
66 |
|
67 |
-
|
68 |
# Command to run your application
|
69 |
CMD ["python", "app.py"]
|
|
|
36 |
python3.8-dev \
|
37 |
&& rm -rf /var/lib/apt/lists/*
|
38 |
|
39 |
+
USER user
|
40 |
# Set Python 3.8 as the default python and pip versions
|
41 |
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.8 1
|
42 |
RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
|
|
|
58 |
# Install additional Python packages
|
59 |
RUN pip install dlib ffmpeg-python
|
60 |
|
|
|
|
|
|
|
61 |
# Set the environment variable to specify the GPU device
|
62 |
ENV CUDA_DEVICE_ORDER=PCI_BUS_ID
|
63 |
ENV CUDA_VISIBLE_DEVICES=0
|
64 |
|
65 |
+
|
66 |
# Command to run your application
|
67 |
CMD ["python", "app.py"]
|