FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04

# タイムゾーン設定
RUN ln -sf /usr/share/zoneinfo/Asia/Tokyo /etc/localtime

# Python3、pip、ffmpegをインストール
RUN apt-get update && \
    apt-get install -y python3 python3-pip ffmpeg && \
    rm -rf /var/lib/apt/lists/*

# pipを最新版にアップグレード
RUN python3 -m pip install --upgrade pip

WORKDIR /app

# requirements.txt をコンテナ内にコピーして、必要なパッケージをインストール
COPY requirements.txt /app/

RUN python3 -m pip install --no-cache-dir -r requirements.txt

COPY . .

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