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"]
|