LivePortrait/docker/Dockerfile

37 lines
1003 B
Docker
Raw Normal View History

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