2024-07-12 07:16:10 +00:00
|
|
|
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
|
|
|
|
|
2024-07-12 07:19:03 +00:00
|
|
|
# Install necessary system packages
|
2024-07-12 07:16:10 +00:00
|
|
|
RUN apt-get update && apt-get install -y \
|
|
|
|
git \
|
|
|
|
cmake \
|
|
|
|
libsm6 \
|
|
|
|
libxext6 \
|
|
|
|
libxrender-dev \
|
|
|
|
libglib2.0-0 \
|
|
|
|
libx11-dev
|
|
|
|
|
2024-07-12 07:19:03 +00:00
|
|
|
# Install required system-level packages
|
2024-07-12 07:16:10 +00:00
|
|
|
RUN apt-get update && apt-get install -y python3-pip python3-dev
|
|
|
|
|
2024-07-12 07:19:03 +00:00
|
|
|
# Upgrade pip
|
2024-07-12 07:16:10 +00:00
|
|
|
RUN pip install --upgrade pip
|
|
|
|
|
2024-07-12 07:19:03 +00:00
|
|
|
# Install required Python packages
|
2024-07-12 07:16:10 +00:00
|
|
|
COPY requirements.txt .
|
|
|
|
RUN pip install -r requirements.txt
|
|
|
|
|
2024-07-12 07:19:03 +00:00
|
|
|
# Set the timezone
|
|
|
|
# Example: Set to Asia/Tokyo
|
2024-07-12 07:16:10 +00:00
|
|
|
RUN ln -sf /usr/share/zoneinfo/Asia/Tokyo /etc/localtime
|
2024-07-12 07:19:03 +00:00
|
|
|
# Set timezone
|
2024-07-12 07:16:10 +00:00
|
|
|
ENV TZ Asia/Tokyo
|
|
|
|
|
|
|
|
RUN apt-get install libopencv-dev -y
|
|
|
|
|
2024-07-12 07:19:03 +00:00
|
|
|
# Set the working directory
|
2024-07-12 07:16:10 +00:00
|
|
|
WORKDIR /LivePortrait
|