Peeble commited on
Commit
8f98cae
·
verified ·
1 Parent(s): b8414e4

Create Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +41 -0
Dockerfile ADDED
@@ -0,0 +1,41 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Use Python 3.12 for better compatibility with current ONNX/CUDA 13 wheels
2
+ FROM python:3.12-slim
3
+
4
+ # Set environment variables
5
+ ENV PYTHONUNBUFFERED=1 \
6
+ GRADIO_SERVER_NAME="0.0.0.0" \
7
+ TORCH_CUDA_ARCH_LIST="8.0;8.6;8.9;9.0"
8
+
9
+ WORKDIR /app
10
+
11
+ # 1. Install System Dependencies (Crucial for torchcodec and FFmpeg)
12
+ RUN apt-get update && apt-get install -y \
13
+ git \
14
+ ffmpeg \
15
+ libnppicc12 \
16
+ libnppig12 \
17
+ libnppidevm-12-0 \
18
+ libgl1 \
19
+ && rm -rf /var/lib/apt/lists/*
20
+
21
+ # 2. Copy requirements
22
+ COPY requirements.txt .
23
+
24
+ # 3. The Optimized PIP Command
25
+ # We use --index-url for PyTorch (GPU) and --extra-index-url for everything else (Gradio/Transformers)
26
+ RUN pip install --no-cache-dir --upgrade pip && \
27
+ pip install --no-cache-dir \
28
+ --index-url https://download.pytorch.org/whl/cu130 \
29
+ --extra-index-url https://pypi.org/simple \
30
+ -r requirements.txt \
31
+ gradio[oauth,mcp]==6.10.0 \
32
+ uvicorn \
33
+ spaces
34
+
35
+ # 4. Copy Application Code
36
+ COPY . .
37
+
38
+ # 5. Expose Gradio Port
39
+ EXPOSE 7860
40
+
41
+ CMD ["python", "app.py"]