environment.yml 253 B

12345678910111213
  1. name: hpc-example
  2. channels:
  3. - defaults
  4. - conda-forge
  5. dependencies:
  6. - pytorch
  7. - pytorch-lightning
  8. - torchvision
  9. - torchmetrics
  10. - setuptools==59.5.0
  11. - lightning-bolts
  12. - jsonargparse