MogensR commited on
Commit
d19b4ae
·
verified ·
1 Parent(s): 1639659

Update Dockerfile

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