edgarhnd commited on
Commit
9e1a1f1
·
1 Parent(s): f4d1369

fix gdown missing error

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -4,7 +4,7 @@ FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
4
  RUN apt-get update && apt-get install -y git git-lfs curl ffmpeg libsm6 libxext6 libgl1-mesa-glx xvfb
5
 
6
  # Install Python dependencies including OpenCV
7
- RUN pip install --no-cache-dir runpod requests opencv-python-headless
8
 
9
  # Set up environment variables for HuggingFace and torch cache
10
  ENV HUGGINGFACE_HUB_CACHE=/workspace/cache
 
4
  RUN apt-get update && apt-get install -y git git-lfs curl ffmpeg libsm6 libxext6 libgl1-mesa-glx xvfb
5
 
6
  # Install Python dependencies including OpenCV
7
+ RUN pip install --no-cache-dir runpod requests opencv-python-headless gdown
8
 
9
  # Set up environment variables for HuggingFace and torch cache
10
  ENV HUGGINGFACE_HUB_CACHE=/workspace/cache