File size: 627 Bytes
9039967
 
227e456
b5d3943
9039967
b5d3943
36f3374
 
9039967
fb236cf
 
36f3374
227e456
9039967
22b00f2
36f3374
227e456
9039967
36f3374
 
22b00f2
46f8dfe
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
# Use an NVIDIA PyTorch container with cuDNN 9.1 support
FROM nvidia/cuda:12.3.2-cudnn9-runtime-ubuntu22.04


# Basic setup
ENV OMP_NUM_THREADS=1
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install -y \
    python3 python3-pip ffmpeg git libsndfile1 \
    # Development tools required to compile native extensions such as llama-cpp-python
    build-essential cmake libopenblas-dev \
    && rm -rf /var/lib/apt/lists/*

# Set up Python environment
COPY requirements.txt .
RUN pip install --upgrade pip && pip install -r requirements.txt

# Copy app files
COPY . /app
WORKDIR /app

CMD ["python3", "app.py"]