fffiloni commited on
Commit
dd1f068
·
verified ·
1 Parent(s): 4e0b16f

Update Dockerfile

Browse files
Files changed (1) hide show
  1. 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==1.10.0 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,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