diffusion-avatars/environment.yml at master · tobias-kirschstein/diffusion-avatars · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# Note: conda dependencies have only 1 "="
#       pip dependencies have 2 "=="
# Fuse pip dependencies together under one " - pip" item
#       Otherwise, only some of the are installed, because conda creates a temporary requirements.txt file
#       only the last -pip section

name: diffusion-avatars

channels:
  - nvidia/label/cuda-11.7.1
  - pytorch
  - conda-forge

dependencies:
  - python=3.8
  - pip
  - jupyter
  - jupyterlab

  # CUDA
  - cudatoolkit
  - cuda-nvcc
  - cuda-libraries-dev  # For cuda_runtime.h
  - ninja

  # Additional PyTorch
  - functorch
  - torchinfo

  - pip:
      # Torch
      - --extra-index-url https://download.pytorch.org/whl/cu117
      - torch==2.0.1+cu117
      - torchvision==0.15.2+cu117
      - einops
      - lightning

      # nvdiffrast & Meshes
      - git+ssh://git@github.com/NVlabs/nvdiffrast.git
      - pyopengl
      - glfw
      - trimesh
      - python-openctm
      - smplx

      # Logging
      - wandb
      - mediapy
      - environs
      - tyro

      # Custom packages for facilitating ML research
      - dreifus==0.1.7
      - elias==0.2.11
      - visage==0.2.5

      # Diffusion
      - accelerate
      - diffusers==0.21.2
      - transformers==4.31.0
      - xformers==0.0.21  # 0.0.20
      - bitsandbytes

      # Evaluation
      - pyfvvdp
      - visage@git+https://github.com/tobias-kirschstein/face-parser.git
      - insightface
      - onnxruntime-gpu
      - torchmetrics