env.yaml 502 B

12345678910111213141516171819202122232425
  1. name: mind-vis
  2. channels:
  3. - pytorch
  4. - defaults
  5. dependencies:
  6. - python=3.8.5
  7. - pip>=20.3
  8. - pip:
  9. - numpy
  10. - matplotlib
  11. - omegaconf==2.1.1
  12. - einops==0.3.0
  13. - torch-fidelity==0.3.0
  14. - --extra-index-url https://download.pytorch.org/whl/cu116
  15. - torch==1.12.1
  16. - torchvision==0.13.1
  17. - Pillow==9.0.1
  18. - timm==0.5.4
  19. - tqdm==4.64.0
  20. - wandb==0.12.21
  21. - torchmetrics==0.9.2
  22. - scikit-image
  23. - pytorch-lightning==1.6.5
  24. - lpips==0.1.4
  25. - -e ./code