docker

Dockerfile

@@ -0,0 +1,48 @@
+FROM nvidia/cuda:12.1.0-devel-ubuntu22.04
+
+# 设置非交互模式
+ENV DEBIAN_FRONTEND=noninteractive
+
+# 安装必要的系统依赖
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    wget \
+    curl \
+    git \
+    ca-certificates \
+ && rm -rf /var/lib/apt/lists/*
+
+# 安装 Miniconda
+RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /tmp/miniconda.sh && \
+    bash /tmp/miniconda.sh -b -p /opt/conda && \
+    rm /tmp/miniconda.sh
+ENV PATH=/opt/conda/bin:$PATH
+
+# 创建 conda 环境 “videograin”，指定 Python 3.10
+RUN conda create -n videograin python=3.10 -y
+
+# 在 “videograin” 环境中安装 PyTorch、CUDA 支持及 Xformers
+RUN conda install -n videograin pytorch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 pytorch-cuda=12.1 -c pytorch -c nvidia -y && \
+    conda run -n videograin pip install --pre -U xformers==0.0.27
+
+# 创建非 root 用户（uid=1000）
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user
+WORKDIR ${HOME}/app
+
+# 将本地 requirements.txt 复制进容器，并在 “videograin” 环境中安装 pip 依赖
+COPY --chown=user:user requirements.txt /tmp/requirements.txt
+RUN conda run -n videograin pip install --no-cache-dir -r /tmp/requirements.txt
+
+# 复制应用代码
+COPY --chown=user:user . ${HOME}/app
+
+# 设置环境变量，使后续运行默认使用 “videograin” 环境中的 Python
+ENV PATH=/opt/conda/envs/videograin/bin:$PATH \
+    PYTHONUNBUFFERED=1
+
+# 如有需要可开放端口（例如使用 gradio 的情况）
+# EXPOSE 7860
+
+# 默认启动命令
+CMD ["python", "app.py"]

environmental.yaml

@@ -1,88 +0,0 @@
-name: videograin
-channels:
-  - pytorch
-  - defaults
-dependencies:
-  - ffmpeg=4.3
-  - numpy=1.24.3
-  - python=3.10.0
-  - pytorch=2.3.1
-  - pytorch-cuda=12.1
-  - pyyaml=6.0.2=
-  - torchvision=0.18.1
-  - yaml=0.2.5
-  - huggingface-hub==0.17.3
-  - xformers==0.0.27
-  - gradio==3.44.4
-  - pip:
-      - absl-py==2.1.0
-      - accelerate==0.24.1
-      - antlr4-python3-runtime==4.9.3
-      - asttokens==2.4.1
-      - bitsandbytes==0.35.4
-      - click==8.1.7
-      - coloredlogs==15.0.1
-      - contourpy==1.2.1
-      - cycler==0.12.1
-      - decorator==5.1.1
-      - decord==0.6.0
-      - diffusers==0.19.0
-      - einops==0.8.0
-      - exceptiongroup==1.2.1
-      - executing==2.0.1
-      - flatbuffers==24.3.25
-      - fonttools==4.53.1
-      - fsspec==2024.6.1
-      - ftfy==6.2.0
-      - grpcio==1.65.0
-      - humanfriendly==10.0
-      - imageio==2.34.2
-      - imageio-ffmpeg==0.5.1
-      - importlib-metadata==8.0.0
-      - ipython==8.26.0
-      - jedi==0.19.1
-      - kiwisolver==1.4.5
-      - markdown==3.6
-      - matplotlib==3.9.1
-      - matplotlib-inline==0.1.7
-      - modelcards==0.1.6
-      - omegaconf==2.3.0
-      - onnxruntime==1.18.1
-      - onnxruntime-gpu==1.18.1
-      - opencv-python==4.10.0.84
-      - packaging==24.1
-      - parso==0.8.4
-      - pexpect==4.9.0
-      - prompt-toolkit==3.0.47
-      - protobuf==4.25.3
-      - psutil==6.0.0
-      - ptyprocess==0.7.0
-      - pure-eval==0.2.2
-      - pyav==12.2.0
-      - pygments==2.18.0
-      - pyparsing==3.1.2
-      - python-dateutil==2.9.0.post0
-      - regex==2024.5.15
-      - safetensors==0.4.3
-      - six==1.16.0
-      - stack-data==0.6.3
-      - tensorboard==2.17.0
-      - tensorboard-data-server==0.7.2
-      - tokenizers==0.14.1
-      - tqdm==4.66.4
-      - traitlets==5.14.3
-      - transformers==4.35.0
-      - triton==2.3.1
-      - wcwidth==0.2.13
-      - werkzeug==3.0.3
-      - zipp==3.19.2
-      - scipy==1.14.0
-      - scikit-learn==1.2.2
-      - nltk==3.8.1
-      - timm==0.6.7
-      - scikit-image==0.24.0
-      - gdown==5.1.0
-      - basicsr-fixed
-      - datasets==2.11.0   # 强制使用旧版本的 datasets，确保与 huggingface-hub==0.17.3 兼容
-
-