Spaces:
Sleeping
Sleeping
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
|