environment.yaml 692 B

1234567891011121314151617181920212223242526272829
  1. name: ldm
  2. channels:
  3. - pytorch
  4. - defaults
  5. dependencies:
  6. - python=3.8.5
  7. - pip=20.3
  8. - cudatoolkit=11.3
  9. - pytorch=1.11.0
  10. - torchvision=0.12.0
  11. - numpy=1.19.2
  12. - pip:
  13. - albumentations==0.4.3
  14. - opencv-python==4.1.2.30
  15. - pudb==2019.2
  16. - imageio==2.9.0
  17. - imageio-ffmpeg==0.4.2
  18. - pytorch-lightning==1.4.2
  19. - omegaconf==2.1.1
  20. - test-tube>=0.7.5
  21. - streamlit>=0.73.1
  22. - einops==0.3.0
  23. - torch-fidelity==0.3.0
  24. - transformers==4.19.2
  25. - torchmetrics==0.6.0
  26. - kornia==0.6
  27. - -e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
  28. - -e git+https://github.com/openai/CLIP.git@main#egg=clip
  29. - -e .