-
Notifications
You must be signed in to change notification settings - Fork 1
/
environment.yaml
35 lines (35 loc) · 823 Bytes
/
environment.yaml
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
name: phylo_diffusion
channels:
- pytorch
- defaults
dependencies:
- python=3.8.5
- pip=20.3
- cudatoolkit=11.3
- pytorch=1.12.1
- torchvision=0.13.1
# - numpy=1.19.2
- pip:
- numpy
- albumentations==0.4.3
- opencv-python==4.1.2.30
- pudb==2019.2
- imageio==2.9.0
- imageio-ffmpeg==0.4.2
- pytorch-lightning==1.4.2
- omegaconf==2.0.0
- test-tube>=0.7.5
- streamlit>=0.73.1
- einops==0.3.0
- torch-fidelity==0.3.0
- transformers==4.35.0
- wandb==0.13.5
- tqdm==4.64.1
- torchmetrics==0.6.0
- einops==0.3.0
- seaborn==0.9.0
- open_clip_torch
- scikit-learn
- -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
- -e git+https://github.com/openai/CLIP.git@main#egg=clip
# - -e .