environment.yml

name: sacflow
channels:
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - pip
  - numpy
  - scipy
  - pandas
  - scikit-image
  - scikit-learn
  - nibabel
  - tqdm
  - pyyaml
  - rich
  - matplotlib
  - tensorboard
  - pip:
      - monai>=1.3.0
      - wandb>=0.17.0
      - einops>=0.7.0
      - SimpleITK>=2.3.0
      - connected-components-3d>=3.12.0
      - medpy>=0.5.2
      # Install torch/torchvision/torchaudio separately for your ROCm version.
      # Example for ROCm 6.2/6.3 clusters, check official PyTorch selector:
      # pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.2