Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
|
@@ -9,7 +9,10 @@ RUN useradd -m -u 1000 user
|
|
| 9 |
USER user
|
| 10 |
|
| 11 |
ENV HOME=/home/user \
|
|
|
|
| 12 |
PATH=/home/user/.local/bin:$PATH \
|
|
|
|
|
|
|
| 13 |
PYTHONPATH=$HOME/app \
|
| 14 |
PYTHONUNBUFFERED=1 \
|
| 15 |
GRADIO_ALLOW_FLAGGING=never \
|
|
@@ -51,7 +54,7 @@ COPY requirements_HF.txt .
|
|
| 51 |
# Install Python dependencies from requirements.txt
|
| 52 |
RUN pip install --upgrade pip
|
| 53 |
RUN pip install -r requirements_HF.txt
|
| 54 |
-
RUN pip install gradio
|
| 55 |
|
| 56 |
USER root
|
| 57 |
# Install basicsr (assuming setup.py is in basicsr directory)
|
|
|
|
| 9 |
USER user
|
| 10 |
|
| 11 |
ENV HOME=/home/user \
|
| 12 |
+
CUDA_HOME=/usr/local/cuda \
|
| 13 |
PATH=/home/user/.local/bin:$PATH \
|
| 14 |
+
LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH} \
|
| 15 |
+
LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:${LIBRARY_PATH} \
|
| 16 |
PYTHONPATH=$HOME/app \
|
| 17 |
PYTHONUNBUFFERED=1 \
|
| 18 |
GRADIO_ALLOW_FLAGGING=never \
|
|
|
|
| 54 |
# Install Python dependencies from requirements.txt
|
| 55 |
RUN pip install --upgrade pip
|
| 56 |
RUN pip install -r requirements_HF.txt
|
| 57 |
+
RUN pip install gradio cupy-cuda11x
|
| 58 |
|
| 59 |
USER root
|
| 60 |
# Install basicsr (assuming setup.py is in basicsr directory)
|