🔧 [feat] Create Dockerfile for environment setup

- Introduced a Dockerfile based on NVIDIA CUDA and Ubuntu with essential packages and Python dependencies for the LivePortrait application. Ensured the system is prepared for high-performance operations with GPU support.
This commit is contained in:
Maki 2024-07-12 16:16:10 +09:00
parent 661fc0eea2
commit 01c6252020

36
docker/Dockerfile Normal file
View File

@ -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