video-retalking容器

Dockerfile文件

# 使用 NVIDIA CUDA 11.1 开发环境作为基础镜像
FROM nvcr.io/nvidia/cuda:11.1.1-cudnn8-runtime-ubuntu20.04

# 设置非交互式安装模式以避免某些命令在构建过程中暂停
ENV DEBIAN_FRONTEND=noninteractive

# 更新软件包列表
RUN apt-get update

# 安装 Python 3.8 和 pip
RUN apt-get install -y  software-properties-common 
RUN apt-get install -y wget
RUN add-apt-repository ppa:deadsnakes/ppa
RUN apt-get install -y  python3.8 && apt-get  install python3-pip -y && apt install git  -y   && apt-get install ffmpeg libsm6 libxext6  -y && apt-get install libgl1 -y && apt-get install -
y g++ && apt-get install -y gcc  &&  apt-get install -y cmake 

# 设置默认的 python 指令为 python3.8
RUN  cp   /usr/bin/python3.8  /usr/bin/python  
RUN  git clone https://github.com/vinthony/video-retalking.git && cd video-retalking && pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 -f https://download.pytorch.org/whl/torch_st
able.html && pip install -r requirements.txt

# 清理缓存以减小镜像体积
RUN apt-get clean && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
# 设置工作目录（可根据需要更改）
WORKDIR /app

# （可选）设置容器启动时执行的命令
CMD ["python"]

##################
 

生产镜像

# 使用 NVIDIA CUDA 11.8 开发环境作为基础镜像
FROM video-retalking:latest 

# 设置非交互式安装模式以避免某些命令在构建过程中暂停
ENV DEBIAN_FRONTEND=noninteractive

# 更新软件包列表
RUN apt-get update

ADD cuda-11.1  /usr/local/cuda-11.1
ADD  video-retalking  /tools
ADD detection_Resnet50_Final.pth /usr/local/lib/python3.8/dist-packages/facexlib/weights/detection_Resnet50_Final.pth
ADD parsing_parsenet.pth  /usr/local/lib/python3.8/dist-packages/facexlib/weights/parsing_parsenet.pth 

# 安装 Python 3.8 和 pip
WORKDIR /app

# （可选）设置容器启动时执行的命令
CMD ["python"]