environment.yml 201 B

12345678910
  1. name: hpc-example
  2. dependencies:
  3. - pytorch
  4. - pytorch-lightning
  5. - torchvision
  6. - torchmetrics
  7. - setuptools==59.5.0
  8. - lightning-bolts
  9. - jsonargparse