SACFlow / environment.yml
sathiiii's picture
Add files using upload-large-folder tool
eea9a47 verified
Raw
History Blame Contribute Delete
645 Bytes
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