R3DPA/environment.yml at main · valeoai/R3DPA · 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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
name: r3dpa
channels:
  - defaults
dependencies:
  - _libgcc_mutex=0.1=main
  - _openmp_mutex=5.1=1_gnu
  - ca-certificates=2025.2.25=h06a4308_0
  - ld_impl_linux-64=2.40=h12ee557_0
  - libffi=3.4.4=h6a678d5_1
  - libgcc-ng=11.2.0=h1234567_1
  - libgomp=11.2.0=h1234567_1
  - libstdcxx-ng=11.2.0=h1234567_1
  - ncurses=6.4=h6a678d5_0
  - openssl=3.0.16=h5eee18b_0
  - pip=25.0=py39h06a4308_0
  - python=3.9.20=he870216_1
  - readline=8.2=h5eee18b_0
  - setuptools=75.8.0=py39h06a4308_0
  - sqlite=3.45.3=h5eee18b_0
  - tk=8.6.14=h39e8969_0
  - tzdata=2025a=h04d1e81_0
  - wheel=0.45.1=py39h06a4308_0
  - xz=5.6.4=h5eee18b_1
  - zlib=1.2.13=h5eee18b_1
  - pip:
      - accelerate==1.1.1
      - antlr4-python3-runtime==4.9.3
      - certifi==2025.1.31
      - charset-normalizer==3.4.1
      - click==8.1.8
      - git+https://github.com/openai/CLIP.git
      - dictdot==1.5.1
      - docker-pycreds==0.4.0
      - einops==0.8.0
      - filelock==3.18.0
      - fsspec==2025.3.0
      - ftfy==6.3.1
      - gitdb==4.0.12
      - gitpython==3.1.44
      - h5py==3.12.1
      - huggingface-hub==0.29.3
      - idna==3.10
      - jinja2==3.1.6
      - markupsafe==3.0.2
      - mpmath==1.3.0
      - networkx==3.2.1
      - numpy==1.26.4
      - nvidia-cublas-cu12==12.1.3.1
      - nvidia-cuda-cupti-cu12==12.1.105
      - nvidia-cuda-nvrtc-cu12==12.1.105
      - nvidia-cuda-runtime-cu12==12.1.105
      - nvidia-cudnn-cu12==8.9.2.26
      - nvidia-cufft-cu12==11.0.2.54
      - nvidia-curand-cu12==10.3.2.106
      - nvidia-cusolver-cu12==11.4.5.107
      - nvidia-cusparse-cu12==12.1.0.106
      - nvidia-nccl-cu12==2.20.5
      - nvidia-nvjitlink-cu12==12.8.93
      - nvidia-nvtx-cu12==12.1.105
      - omegaconf==2.3.0
      - packaging==24.2
      - pillow==11.1.0
      - platformdirs==4.3.7
      - protobuf==5.29.4
      - psutil==7.0.0
      - pyyaml==6.0.2
      - regex==2024.11.6
      - requests==2.32.3
      - safetensors==0.5.3
      - sentry-sdk==2.24.0
      - setproctitle==1.3.5
      - six==1.17.0
      - smmap==5.0.2
      - sympy==1.13.3
      - timm==1.0.11
      - tokenizers==0.20.3
      - torch==2.3.1
      - torchvision==0.18.1
      - tqdm==4.67.1
      - transformers==4.46.3
      - triton==2.3.1
      - typing-extensions==4.12.2
      - urllib3==2.3.0
      - wandb==0.18.7
      - wcwidth==0.2.13
      - tensorboard==2.19.0
      - dictdot==1.5.1
      - pandas==2.3.2
      - matplotlib==3.9.4
      - scipy==1.13.1
      - pytorch-lightning==1.4.2
      - torchmetrics==0.5.0
      - easydict==1.13
      # - ninja==1.13.0
      #- git+https://github.com/mit-han-lab/torchsparse.git@v1.4.0