Update Dockerfile
Browse files- Dockerfile +6 -2
Dockerfile
CHANGED
|
@@ -43,7 +43,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
| 43 |
build-essential gcc g++ pkg-config \
|
| 44 |
libffi-dev libssl-dev libc6-dev \
|
| 45 |
libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 libgomp1 \
|
| 46 |
-
libopus0 libvpx7 libsrtp2-1 libx264-dev libx265-dev libmp3lame-dev
|
|
|
|
| 47 |
rm -rf /var/lib/apt/lists/*
|
| 48 |
|
| 49 |
# ---- Install FFmpeg with Full Codec Support ----
|
|
@@ -55,6 +56,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
| 55 |
# ---- Python Bootstrap ----
|
| 56 |
RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
|
| 57 |
|
|
|
|
|
|
|
|
|
|
| 58 |
# ---- Install PyTorch (CUDA 12.1) ----
|
| 59 |
RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 \
|
| 60 |
torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 && \
|
|
@@ -92,7 +96,7 @@ COPY --chown=user:user . .
|
|
| 92 |
|
| 93 |
# ---- Healthcheck ----
|
| 94 |
HEALTHCHECK --interval=30s --timeout=8s --retries=3 CMD \
|
| 95 |
-
python3 -c "import torch, streamlit; print('torch', torch.__version__, '| cuda', torch.cuda.is_available()); print('streamlit', streamlit.__version__); import subprocess; subprocess.run(['ffmpeg', '-version'], check=False)"
|
| 96 |
|
| 97 |
# ---- Runtime Startup ----
|
| 98 |
USER user
|
|
|
|
| 43 |
build-essential gcc g++ pkg-config \
|
| 44 |
libffi-dev libssl-dev libc6-dev \
|
| 45 |
libgl1-mesa-glx libglib2.0-0 libsm6 libxext6 libxrender1 libgomp1 \
|
| 46 |
+
libopus0 libvpx7 libsrtp2-1 libx264-dev libx265-dev libmp3lame-dev \
|
| 47 |
+
libblas-dev liblapack-dev gfortran && \
|
| 48 |
rm -rf /var/lib/apt/lists/*
|
| 49 |
|
| 50 |
# ---- Install FFmpeg with Full Codec Support ----
|
|
|
|
| 56 |
# ---- Python Bootstrap ----
|
| 57 |
RUN python3 -m pip install --no-cache-dir --upgrade pip setuptools wheel
|
| 58 |
|
| 59 |
+
# ---- Install NumPy ----
|
| 60 |
+
RUN python3 -m pip install --no-cache-dir numpy==1.24.4
|
| 61 |
+
|
| 62 |
# ---- Install PyTorch (CUDA 12.1) ----
|
| 63 |
RUN python3 -m pip install --no-cache-dir --index-url https://download.pytorch.org/whl/cu121 \
|
| 64 |
torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 && \
|
|
|
|
| 96 |
|
| 97 |
# ---- Healthcheck ----
|
| 98 |
HEALTHCHECK --interval=30s --timeout=8s --retries=3 CMD \
|
| 99 |
+
python3 -c "import torch, streamlit, numpy; print('torch', torch.__version__, '| cuda', torch.cuda.is_available()); print('streamlit', streamlit.__version__); print('numpy', numpy.__version__); import subprocess; subprocess.run(['ffmpeg', '-version'], check=False)"
|
| 100 |
|
| 101 |
# ---- Runtime Startup ----
|
| 102 |
USER user
|