diff --git a/docker/Dockerfile b/docker/Dockerfile new file mode 100644 index 0000000..e8f9dc7 --- /dev/null +++ b/docker/Dockerfile @@ -0,0 +1,36 @@ +FROM nvidia/cuda:12.0.1-cudnn8-runtime-ubuntu22.04 +# FROM nvidia/cuda:12.5.0-devel-ubuntu22.04 +# FROM pytorch/pytorch:2.3.1-cuda12.1-cudnn8-devel + +WORKDIR /LivePortrait + +# # 必要なシステムパッケージをインストールする +RUN apt-get update && apt-get install -y \ + git \ + cmake \ + libsm6 \ + libxext6 \ + libxrender-dev \ + libglib2.0-0 \ + libx11-dev + +# システムレベルで必要なパッケージをインストールします +RUN apt-get update && apt-get install -y python3-pip python3-dev + +# pip をアップグレードする +RUN pip install --upgrade pip + +# 必要な Python パッケージをインストールする +COPY requirements.txt . +RUN pip install -r requirements.txt + +# タイムゾーンを設定する +# 例としてAsia/Tokyoを設定 +RUN ln -sf /usr/share/zoneinfo/Asia/Tokyo /etc/localtime +# タイムゾーンを設定 +ENV TZ Asia/Tokyo + +RUN apt-get install libopencv-dev -y + +# 作業ディレクトリを設定する +WORKDIR /LivePortrait