Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -1
Dockerfile
CHANGED
@@ -54,9 +54,10 @@ COPY requirements_HF.txt .
|
|
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
|
58 |
|
59 |
USER root
|
|
|
60 |
# Install basicsr (assuming setup.py is in basicsr directory)
|
61 |
RUN python basicsr/setup.py develop
|
62 |
|
|
|
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
|
58 |
|
59 |
USER root
|
60 |
+
RUN pip install cupy-cuda11x
|
61 |
# Install basicsr (assuming setup.py is in basicsr directory)
|
62 |
RUN python basicsr/setup.py develop
|
63 |
|