iris/environment.yml at main · facebookresearch/iris · 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
name: iris
channels:
  - defaults
dependencies:
  - python=3.8
  - pip
  - pip:
      # Core Python libraries
      - numpy==1.24.4
      - torch==1.13.1+cu117
      - torchvision==0.14.1+cu117
      - --extra-index-url https://download.pytorch.org/whl/cu117
      - pytorch-lightning==1.9.0
      - mitsuba==3.5.0
      - drjit==0.4.4

      # Image processing and visualization
      - opencv-python
      - imageio
      - imageio-ffmpeg
      - matplotlib
      - pillow

      # Utilities
      - tqdm==4.66.1
      - argparse==1.4.0
      - omegaconf==2.3.0
      - scipy==1.10.1
      - scikit-image==0.21.0
      - trimesh==3.21.4
      - pyquaternion==0.9.9

      # Logging and monitoring
      - tensorboardx==2.6.2.2
      - wandb==0.16.1

      # File and path handling
      - pathlib==1.0.1

      # Optional dependencies for CRF and plotting
      - lpips==0.1.4
      - matplotlib-inline==0.1.6
      - git+https://github.com/sbarratt/torch_interpolations.git