a | b/environment.yaml | ||
---|---|---|---|
1 | name: venv |
||
2 | channels: |
||
3 | - pytorch |
||
4 | - defaults |
||
5 | - anaconda |
||
6 | dependencies: |
||
7 | - python=3.9 |
||
8 | - cudatoolkit |
||
9 | - pip |
||
10 | - pip: |
||
11 | - torch==2.0.0 |
||
12 | - torchaudio |
||
13 | - torchvision |
||
14 | - huggingface-hub==0.18.0 |
||
15 | - matplotlib==3.7.0 |
||
16 | - psutil==5.9.4 |
||
17 | - iopath |
||
18 | - pyyaml==6.0 |
||
19 | - regex==2022.10.31 |
||
20 | - tokenizers==0.13.2 |
||
21 | - tqdm==4.64.1 |
||
22 | - transformers==4.30.0 |
||
23 | - timm==0.6.13 |
||
24 | - webdataset==0.2.48 |
||
25 | - omegaconf==2.3.0 |
||
26 | - opencv-python==4.7.0.72 |
||
27 | - decord==0.6.0 |
||
28 | - peft==0.2.0 |
||
29 | - sentence-transformers |
||
30 | - gradio==3.47.1 |
||
31 | - accelerate==0.20.3 |
||
32 | - bitsandbytes==0.37.0 |
||
33 | - scikit-image |
||
34 | - visual-genome |
||
35 | - wandb |