R2MBench/environment.yml at main · AMAP-ML/R2MBench · 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
name: mem-benchmark-full-eval
channels:
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - pip
  - pip:
      - --extra-index-url
      - https://download.pytorch.org/whl/cu124
      - torch==2.6.0+cu124
      - torchvision==0.21.0+cu124
      - numpy==2.2.6
      - opencv-python==4.13.0.92
      - pillow==12.1.1
      - scipy==1.15.3
      - scikit-image==0.25.2
      - scikit-learn==1.7.2
      - decord==0.6.0
      - lpips==0.1.4
      - tqdm==4.67.3
      - matplotlib==3.10.8
      - pandas==2.3.3
      - requests==2.32.5
      - python-dotenv==1.2.2
      - cvxpy==1.7.5
      - einops==0.8.2
      - transformers==4.57.6
      - huggingface-hub==0.23.0
      - safetensors==0.7.0
      - omegaconf==2.3.0
      - hydra-core==1.3.2
      - pycocotools==2.0.11
      - supervision==0.27.0.post1
      - accelerate==1.1.0
      - timm==1.0.25
      - addict==2.4.0
      - yapf==0.43.0
      - openai>=1.0.0
      - git+https://github.com/openai/CLIP.git