environment.yml 717 B

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. name: gmpi
  2. channels:
  3. - defaults
  4. - pytorch
  5. - conda-forge
  6. dependencies:
  7. - python=3.7
  8. - numpy
  9. - pandas
  10. - scipy
  11. - tqdm
  12. - h5py
  13. - jupyterlab
  14. - matplotlib
  15. - joblib
  16. - lz4
  17. - pylint
  18. - pillow=6.1
  19. - pytorch=1.9.1
  20. - torchvision=0.10.1
  21. - torchaudio==0.9.1
  22. - cudatoolkit=10.2
  23. - tensorboard
  24. - pip
  25. - pip:
  26. - setuptools==59.5.0
  27. - lpips
  28. - yacs
  29. - scikit-image
  30. - scikit-video
  31. - imageio
  32. - opencv-python
  33. - black
  34. - trimesh
  35. - flake8
  36. - flake8-bugbear
  37. - flake8-comprehensions
  38. - clean-fid
  39. - torch_ema
  40. - torch-fidelity
  41. - mrcfile
  42. - dill
  43. - click
  44. - psutil
  45. - lazy
  46. - imageio-ffmpeg==0.4.3
  47. - pyspng==0.1.0
  48. - PyMCubes