Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -55,7 +55,7 @@ WORKDIR $HOME/app
|
|
| 55 |
# Upgrade pip and install numpy<2 first, then PyTorch (with correct CUDA version)
|
| 56 |
RUN pip install --no-cache-dir --upgrade pip && \
|
| 57 |
pip install --no-cache-dir "numpy<2" && \
|
| 58 |
-
pip install --no-cache-dir torch
|
| 59 |
python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA Available:', torch.cuda.is_available()); print('Device:', torch.cuda.get_device_name(0))"
|
| 60 |
|
| 61 |
# Clone the repository
|
|
@@ -67,7 +67,7 @@ COPY requirements_HF.txt .
|
|
| 67 |
|
| 68 |
# Install Python dependencies
|
| 69 |
RUN pip install --no-cache-dir -r requirements_HF.txt
|
| 70 |
-
RUN pip install --no-cache-dir gradio ffmpeg-python dlib-
|
| 71 |
RUN pip install basicsr
|
| 72 |
|
| 73 |
# Set CUDA device settings
|
|
|
|
| 55 |
# Upgrade pip and install numpy<2 first, then PyTorch (with correct CUDA version)
|
| 56 |
RUN pip install --no-cache-dir --upgrade pip && \
|
| 57 |
pip install --no-cache-dir "numpy<2" && \
|
| 58 |
+
pip install --no-cache-dir torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113 && \
|
| 59 |
python -c "import torch; print('PyTorch version:', torch.__version__); print('CUDA Available:', torch.cuda.is_available()); print('Device:', torch.cuda.get_device_name(0))"
|
| 60 |
|
| 61 |
# Clone the repository
|
|
|
|
| 67 |
|
| 68 |
# Install Python dependencies
|
| 69 |
RUN pip install --no-cache-dir -r requirements_HF.txt
|
| 70 |
+
RUN pip install --no-cache-dir gradio ffmpeg-python dlib-bin
|
| 71 |
RUN pip install basicsr
|
| 72 |
|
| 73 |
# Set CUDA device settings
|