VenkateshRoshan commited on
Commit
1b8e514
·
1 Parent(s): aa852f0

dockerfile update

Browse files
Files changed (1) hide show
  1. dockerfile_hf +11 -0
dockerfile_hf CHANGED
@@ -6,6 +6,17 @@ COPY app_hf.py .
6
  COPY src/ .
7
  COPY requirements.txt .
8
 
 
 
 
 
 
 
 
 
 
 
 
9
  RUN pip install --no-cache-dir --upgrade pip
10
  RUN pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
11
  RUN pip install --no-cache-dir -r requirements.txt
 
6
  COPY src/ .
7
  COPY requirements.txt .
8
 
9
+ # Install system dependencies
10
+ RUN apt-get update && \
11
+ DEBIAN_FRONTEND=noninteractive apt-get install -y \
12
+ python3-pip \
13
+ python3-dev \
14
+ git \
15
+ libomp-dev \
16
+ libopenblas-dev \
17
+ libblas-dev \
18
+ && rm -rf /var/lib/apt/lists/*
19
+
20
  RUN pip install --no-cache-dir --upgrade pip
21
  RUN pip install --no-cache-dir torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121
22
  RUN pip install --no-cache-dir -r requirements.txt