environment.yml

name: talking_gaussian
channels:
  - pytorch
  - pytorch3d
  - nvidia
  - conda-forge
  - defaults
dependencies:
  - cudatoolkit=11.3
  - plyfile=0.8.1
  - python=3.10.15
  - pip
  - pytorch3d
  - pytorch-cuda=12.1
  - pytorch=1.12.1
  - torchaudio=0.12.1
  - torchvision=0.13.1
  - tqdm
  - ffmpeg
  - openh264
  - pip:
    - ./submodules/diff-gaussian-rasterization
    - ./submodules/simple-knn
    - ./gridencoder
    - numpy
    - pillow
    - scipy
    - tensorboard==2.14.1
    - opencv-python
    - tensorboardX

    - pandas
    - tqdm
    - matplotlib
    - PyMCubes==0.1.4
    - rich
    - packaging
    - scikit-learn

    - face_alignment
    - python_speech_features
    - ninja
    - numba
    - resampy
    - psutil
    - pyaudio
    - soundfile
    - configargparse

    - lpips
    - imageio-ffmpeg