GenSeg-Baselines / code /envs /nnunet.yml
MaybeRichard's picture
Upload code/envs/nnunet.yml with huggingface_hub
a654119 verified
Raw
History Blame Contribute Delete
468 Bytes
# Isolated env for nnU-Net v2 reference baseline (own CLI + dataset format).
# Run separately; reported as a reference row, not in the controlled-augmentation table.
name: nnunet
channels: [conda-forge]
dependencies:
- python=3.11
- pip
- pip:
# torch >=2.1.2 (nnU-Net excludes 2.9.x); install matching your CUDA:
# pip install torch torchvision --index-url https://download.pytorch.org/whl/cu121
- nnunetv2
- simpleitk
- nibabel